diff --git a/pytato/distributed.py b/pytato/distributed.py
index a7149c82933a37664d2dc6d0647861e2dd70c9ff..6c52ccfa7cbb1992777a9076919221398c626b6c 100644
--- a/pytato/distributed.py
+++ b/pytato/distributed.py
@@ -956,11 +956,11 @@ def number_distributed_tags(
     tags = frozenset({
             recv.comm_tag
             for part in partition.parts.values()
-            for name, recv in part.input_name_to_recv_node.items()
+            for recv in part.input_name_to_recv_node.values()
             } | {
             send.comm_tag
             for part in partition.parts.values()
-            for name, send in part.output_name_to_send_node.items()})
+            for send in part.output_name_to_send_node.values()})
 
     from mpi4py import MPI