diff --git a/distributed/comm/ucx.py b/distributed/comm/ucx.py
index 72c3c59a63..386c181017 100644
--- a/distributed/comm/ucx.py
+++ b/distributed/comm/ucx.py
@@ -239,9 +239,7 @@ async def read(self, deserializers=("cuda", "dask", "pickle", "error")):
             else:
                 # Recv frames
                 frames = [
-                    device_array(each_size)
-                    if is_cuda
-                    else host_array(each_size)
+                    device_array(each_size) if is_cuda else host_array(each_size)
                     for is_cuda, each_size in zip(cuda_frames, sizes)
                 ]
                 recv_frames = [