diff --git a/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint1.py b/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint1.py
index c9172e74f28a354eb8aa1d83337bf4676cf508a9..1f0cec309c54c4e25b36a1595b53eadea7382705 100644
--- a/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint1.py
+++ b/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint1.py
@@ -30,7 +30,7 @@ import numpy as np
 from paddle.io import Dataset, BatchSampler, DataLoader
 
 from paddle.fluid.tests.unittests.auto_checkpoint_utils import AutoCheckpointBase, get_logger
-from paddle.fluid.tests.unittests.test_auto_checkpoint import AutoCheckPointACLBase
+from test_auto_checkpoint import AutoCheckPointACLBase
 
 paddle.enable_static()
 logger = get_logger()
diff --git a/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint2.py b/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint2.py
index 22b3c15053a41c3ed3f89de4dca8f6b597f047a1..846b31d3fa8d13e7de99519d672e1ed3fe522b55 100644
--- a/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint2.py
+++ b/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint2.py
@@ -30,7 +30,7 @@ import numpy as np
 from paddle.io import Dataset, BatchSampler, DataLoader
 
 from paddle.fluid.tests.unittests.auto_checkpoint_utils import AutoCheckpointBase, get_logger
-from paddle.fluid.tests.unittests.test_auto_checkpoint import AutoCheckPointACLBase
+from test_auto_checkpoint import AutoCheckPointACLBase
 
 paddle.enable_static()
 logger = get_logger()
diff --git a/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint3.py b/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint3.py
index 8d847fe9704e0e209a2903cdfb0e7c384c000dea..d6f11df20b29d5170a9e35b507488be44289c20d 100644
--- a/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint3.py
+++ b/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint3.py
@@ -30,7 +30,7 @@ import numpy as np
 from paddle.io import Dataset, BatchSampler, DataLoader
 
 from paddle.fluid.tests.unittests.auto_checkpoint_utils import AutoCheckpointBase, get_logger
-from paddle.fluid.tests.unittests.test_auto_checkpoint import AutoCheckPointACLBase
+from test_auto_checkpoint import AutoCheckPointACLBase
 
 paddle.enable_static()
 logger = get_logger()
diff --git a/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint_dist_basic.py b/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint_dist_basic.py
index c0aa13aa03f0d07a9ecb79963ead612307220f03..4949f4728eaf34313f2f7e0088d2a9fdd8453b10 100644
--- a/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint_dist_basic.py
+++ b/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint_dist_basic.py
@@ -30,7 +30,7 @@ import numpy as np
 from paddle.io import Dataset, BatchSampler, DataLoader
 
 from paddle.fluid.tests.unittests.auto_checkpoint_utils import AutoCheckpointBase, get_logger
-from paddle.fluid.tests.unittests.test_auto_checkpoint import AutoCheckPointACLBase
+from test_auto_checkpoint import AutoCheckPointACLBase
 
 paddle.enable_static()
 logger = get_logger()
diff --git a/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint_multiple.py b/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint_multiple.py
index da7f2af169d92038a70b8644a525b37aa9ad0f78..65de718a487da50f0c1e3fbe52e0f140563c58f1 100644
--- a/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint_multiple.py
+++ b/python/paddle/fluid/tests/unittests/collective/fleet/test_auto_checkpoint_multiple.py
@@ -30,7 +30,7 @@ import numpy as np
 from paddle.io import Dataset, BatchSampler, DataLoader
 
 from paddle.fluid.tests.unittests.auto_checkpoint_utils import AutoCheckpointBase, get_logger
-from paddle.fluid.tests.unittests.test_auto_checkpoint import AutoCheckPointACLBase
+from test_auto_checkpoint import AutoCheckPointACLBase
 
 paddle.enable_static()
 logger = get_logger()