diff --git a/src/earthkit/workflows/__init__.py b/src/earthkit/workflows/__init__.py
index 86078b9f..7f211331 100644
--- a/src/earthkit/workflows/__init__.py
+++ b/src/earthkit/workflows/__init__.py
@@ -18,7 +18,6 @@
 from . import fluent, mark
 from .graph import Graph, deduplicate_nodes
 from .graph.export import deserialise, serialise
-from .visualise import visualise as _visualise_fn
 
 
 class Cascade:
@@ -33,6 +32,8 @@ def from_actions(cls, actions):
         return cls(deduplicate_nodes(graph))
 
     def visualise(self, *args, **kwargs):
+        from .visualise import visualise as _visualise_fn
+
         return _visualise_fn(self._graph, *args, **kwargs)
 
     def __add__(self, other: "Cascade") -> "Cascade":