diff --git a/pyzx/graph/__init__.py b/pyzx/graph/__init__.py
index 91d72fa2..274c5d5a 100644
--- a/pyzx/graph/__init__.py
+++ b/pyzx/graph/__init__.py
@@ -26,5 +26,6 @@
 
 from .graph import Graph
 from .scalar import Scalar
-from .base import EdgeType, VertexType, toggle_edge, vertex_is_zx, toggle_vertex
+from .base import EdgeType, VertexType
+from ..utils import toggle_vertex, toggle_edge, vertex_is_zx
 from .diff import GraphDiff