diff --git a/setup.py b/setup.py
index c6f6093..a4bafdb 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,7 @@
     description="A simple, easy-to-hack GraphRAG implementation",
     long_description=long_description,
     long_description_content_type="text/markdown",
-    packages=["nano_graphrag"],
+    packages=["nano_graphrag", "nano_graphrag._storage"],
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: MIT License",