diff --git a/python/paddle/amp/grad_scaler.py b/python/paddle/amp/grad_scaler.py
index d7f99f1a90563..5fd8ff20a2041 100644
--- a/python/paddle/amp/grad_scaler.py
+++ b/python/paddle/amp/grad_scaler.py
@@ -37,8 +37,8 @@
 
 if TYPE_CHECKING:
     from paddle import Tensor
+    from paddle.optimizer.optimizer import Optimizer
     from paddle.static.amp.decorator import OptimizerWithMixedPrecision
-    from python.paddle.optimizer.optimizer import Optimizer
 
     class _ScaleStateDict(TypedDict):
         scale: Tensor