diff --git a/src/py_d2/__init__.py b/src/py_d2/__init__.py
index d6b2d56..4e74756 100644
--- a/src/py_d2/__init__.py
+++ b/src/py_d2/__init__.py
@@ -1,8 +1,9 @@
 from .connection import D2Connection
 from .connection import Direction
 from .diagram import D2Diagram
-from .shape import D2Shape, Shape
+from .shape import D2Shape
 from .shape import D2Text
+from .shape import Shape
 from .style import D2Style