diff --git a/imperative/python/test/unit/test_distributed.py b/imperative/python/test/unit/test_distributed.py
index dd94ac05bc05594c3697b518b03d6c928ddbd401..dd5add7e5c529fa87d492dbb5a735777b14c1ee5 100644
--- a/imperative/python/test/unit/test_distributed.py
+++ b/imperative/python/test/unit/test_distributed.py
@@ -73,7 +73,6 @@ def test_init_process_group():
             assert p.exitcode == 0
 
     check("nccl")
-    check("ucx")
 
 
 @pytest.mark.skipif(
diff --git a/python_module/test/unit/distributed/test_functional.py b/python_module/test/unit/distributed/test_functional.py
index f4adcbd98de6ed21b8fde3c3d28f2194eb7d9706..8bbaf7e05f427169f918e1c05bbc1e40134cafd1 100644
--- a/python_module/test/unit/distributed/test_functional.py
+++ b/python_module/test/unit/distributed/test_functional.py
@@ -64,7 +64,7 @@ def test_reduce_sum():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     for shape in [(2, 3), (8, 10), (99, 77)]:
-        for backend in ["nccl", "ucx"]:
+        for backend in ["nccl"]:
             check(shape, backend)
 
 
@@ -106,7 +106,7 @@ def test_gather():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     for shape in [(2, 3), (8, 10), (99, 77)]:
-        for backend in ["nccl", "ucx"]:
+        for backend in ["nccl"]:
             check(shape, backend)
 
 
@@ -144,7 +144,7 @@ def test_broadcast():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     for shape in [(2, 3), (8, 10), (99, 77)]:
-        for backend in ["nccl", "ucx"]:
+        for backend in ["nccl"]:
             check(shape, backend)
 
 
@@ -186,7 +186,7 @@ def test_scatter():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     for shape in [(2, 3), (8, 10), (100, 77)]:
-        for backend in ["nccl", "ucx"]:
+        for backend in ["nccl"]:
             check(shape, backend)
 
 
@@ -226,7 +226,7 @@ def test_all_to_all():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     for shape in [(2, 3), (8, 10), (100, 77)]:
-        for backend in ["nccl", "ucx"]:
+        for backend in ["nccl"]:
             check(shape, backend)
 
 
@@ -265,7 +265,7 @@ def test_all_gather():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     for shape in [(2, 3), (8, 10), (99, 77)]:
-        for backend in ["nccl", "ucx"]:
+        for backend in ["nccl"]:
             check(shape, backend)
 
 
@@ -308,7 +308,7 @@ def test_reduce_scatter_sum():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     for shape in [(2, 4), (8, 10), (88, 44)]:
-        for backend in ["nccl", "ucx"]:
+        for backend in ["nccl"]:
             check(shape, backend)
 
 
@@ -347,7 +347,7 @@ def test_all_reduce_sum():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     for shape in [(2, 3), (8, 10), (99, 77)]:
-        for backend in ["nccl", "ucx"]:
+        for backend in ["nccl"]:
             check(shape, backend)
 
 
@@ -386,7 +386,7 @@ def test_all_reduce_max():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     for shape in [(2, 3), (8, 10), (99, 77)]:
-        for backend in ["nccl", "ucx"]:
+        for backend in ["nccl"]:
             check(shape, backend)
 
 
@@ -425,7 +425,7 @@ def test_all_reduce_min():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     for shape in [(2, 3), (8, 10), (99, 77)]:
-        for backend in ["nccl", "ucx"]:
+        for backend in ["nccl"]:
             check(shape, backend)
 
 
@@ -463,5 +463,5 @@ def test_bcast_param():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     for shape in [(2, 3), (8, 10), (99, 77)]:
-        for backend in ["nccl", "ucx"]:
+        for backend in ["nccl"]:
             check(shape, backend)
diff --git a/python_module/test/unit/distributed/test_util.py b/python_module/test/unit/distributed/test_util.py
index 30926b7910f7c78d871c248aeec636d87c06369e..04b34244176885af09c92016574161a75d175468 100644
--- a/python_module/test/unit/distributed/test_util.py
+++ b/python_module/test/unit/distributed/test_util.py
@@ -102,7 +102,6 @@ def test_init_process_group():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     check("nccl")
-    check("ucx")
 
 
 @pytest.mark.skipif(