diff --git a/python/paddle/fluid/tests/unittests/xpu/process_group_bkcl.py b/python/paddle/fluid/tests/unittests/xpu/process_group_bkcl.py
index 5ea3845c0bd18262264a85e2fc02fe85a54a79e9..2317e38cb28d0d7a16d8e92ab844645adf0f5f64 100644
--- a/python/paddle/fluid/tests/unittests/xpu/process_group_bkcl.py
+++ b/python/paddle/fluid/tests/unittests/xpu/process_group_bkcl.py
@@ -18,6 +18,7 @@ import numpy as np
 import sys
 
 import paddle
+import paddle.distributed as dist
 from paddle.fluid.framework import _test_eager_guard
 from paddle.fluid.dygraph.parallel import ParallelEnv