diff --git a/grudge/op.py b/grudge/op.py
index c6bcb9b1a..b87f6b0ef 100644
--- a/grudge/op.py
+++ b/grudge/op.py
@@ -122,7 +122,7 @@
 from grudge.trace_pair import (
     bdry_trace_pair,
     bv_trace_pair,
-    connected_ranks,
+    # connected_ranks,
     cross_rank_trace_pairs,
     cross_rank_inter_volume_trace_pairs,
     inter_volume_trace_pairs,
@@ -139,7 +139,6 @@
     "bdry_trace_pair",
     "bv_trace_pair",
     "connected_parts",
-    "connected_ranks",
     "cross_rank_inter_volume_trace_pairs",
     "cross_rank_trace_pairs",
     "elementwise_integral",
diff --git a/grudge/trace_pair.py b/grudge/trace_pair.py
index 10c2c06b5..098001601 100644
--- a/grudge/trace_pair.py
+++ b/grudge/trace_pair.py
@@ -72,6 +72,7 @@
 
 from arraycontext import (
     ArrayContainer,
+    ArrayContext,
     ArrayOrContainer,
     dataclass_array_container,
     flatten,