diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py
index b994379d779b86c0de1acbfc0eb43b22ea7e9e50..8e3737b72d6fd71d5fe69852dcb634780e0d2a1d 100644
--- a/python/paddle/fluid/framework.py
+++ b/python/paddle/fluid/framework.py
@@ -4156,7 +4156,7 @@ class Block:
             )
         else:
             from paddle.fluid.dygraph.base import param_guard
-            from .layers.utils import flatten
+            from paddle.utils import flatten
 
             def pass_stop_gradient(ins, outs):
                 """