diff --git a/setup.py b/setup.py
index e9cceec1..3d5c62ff 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@ def parse_requirements(file_name: str) -> List[str]:
 
 setuptools.setup(
     name="llm-guard",
-    version="0.0.1",
+    version="0.0.2",
     author="Laiyer.ai",
     author_email="hello@laiyer.ai",
     description=(
@@ -34,4 +34,7 @@ def parse_requirements(file_name: str) -> List[str]:
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/laiyer-ai/llm-guard",
+    package_data={
+        "llm_guard": ["**/*.json"],
+    },
 )