diff --git a/cubed/core/plan.py b/cubed/core/plan.py
index 3965a46fa..03370f862 100644
--- a/cubed/core/plan.py
+++ b/cubed/core/plan.py
@@ -13,7 +13,6 @@
 import networkx as nx
 
 from cubed.core.optimization import is_input_array, multiple_inputs_optimize_dag
-from cubed.diagnostics.colors import APRICOT, LAVENDER, RED
 from cubed.primitive.blockwise import BlockwiseSpec
 from cubed.primitive.types import PrimitiveOperation
 from cubed.runtime.pipeline import visit_node_generations
@@ -612,6 +611,8 @@ def visualize(
         show_hidden=False,
         engine: Literal["cytoscape", "graphviz"] | None = None,
     ):
+        from cubed.diagnostics.colors import APRICOT, LAVENDER, RED
+
         if engine == "cytoscape":
             return self.visualize_cytoscape(
                 filename,