diff --git a/python/paddle/distributed/communication/stream/all_reduce.py b/python/paddle/distributed/communication/stream/all_reduce.py
index f94422f4bd0a612b69ed4d58be3278f5b4f2ddd4..e4cfa6d3218c23e8cb2fbd43aff23cfe92e85aa4 100644
--- a/python/paddle/distributed/communication/stream/all_reduce.py
+++ b/python/paddle/distributed/communication/stream/all_reduce.py
@@ -12,8 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.distributed.collective as collective
 import paddle.fluid.framework as framework
+from paddle.distributed import collective
 
 
 def _all_reduce_in_dygraph(tensor, op, group, sync_op, use_calc_stream):
diff --git a/python/paddle/distributed/communication/stream/recv.py b/python/paddle/distributed/communication/stream/recv.py
index b225f64b8b4d2508df1b10b64053f689c7bdb20e..8b77fa51b0de079e760446609023fdfbc57066f3 100644
--- a/python/paddle/distributed/communication/stream/recv.py
+++ b/python/paddle/distributed/communication/stream/recv.py
@@ -12,8 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.distributed.collective as collective
 import paddle.fluid.framework as framework
+from paddle.distributed import collective
 
 
 def _recv_in_dygraph(tensor, src, group, sync_op, use_calc_stream):
diff --git a/python/paddle/distributed/communication/stream/send.py b/python/paddle/distributed/communication/stream/send.py
index fa052734c7ee7bcb256f90322845c4d34297b541..d46f0f768cc03ccd513c411c67b68f5ac4f9f046 100644
--- a/python/paddle/distributed/communication/stream/send.py
+++ b/python/paddle/distributed/communication/stream/send.py
@@ -12,8 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.distributed.collective as collective
 import paddle.fluid.framework as framework
+from paddle.distributed import collective
 
 
 def _send_in_dygraph(tensor, dst, group, sync_op, use_calc_stream):