diff --git a/src/py_d2/__init__.py b/src/py_d2/__init__.py
index 7663114..90e2868 100644
--- a/src/py_d2/__init__.py
+++ b/src/py_d2/__init__.py
@@ -1,7 +1,7 @@
 from .connection import D2Connection
 from .connection import Direction
 from .diagram import D2Diagram
-from .shape import D2Shape
+from .shape import D2Shape, Shape
 from .shape import D2Text
 from .style import D2Style
 
@@ -13,4 +13,5 @@
     "D2Shape",
     "D2Text",
     "D2Style",
+    "Shape"
 ]