diff --git a/python/paddle/fluid/tests/unittests/test_IntermediateLayerGetter.py b/python/paddle/fluid/tests/unittests/test_IntermediateLayerGetter.py
index 6a4f5de658800..52aafd18710c2 100644
--- a/python/paddle/fluid/tests/unittests/test_IntermediateLayerGetter.py
+++ b/python/paddle/fluid/tests/unittests/test_IntermediateLayerGetter.py
@@ -15,7 +15,7 @@
 import paddle
 import unittest
 import random
-from paddle.vision.models import IntermediateLayerGetter
+from paddle.vision.models.utils import IntermediateLayerGetter
 
 
 class TestBase: