diff --git a/pyproject.toml b/pyproject.toml
index d3574eb..ea8eabe 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
-name = "validator-template"
-version = "0.0.0"
+name = "llamaguard-7b
+version = "0.0.1"
 description = "Template repo for Guardrails Hub validators."
 authors = [
     {name = "Guardrails AI", email = "contact@guardrailsai.com"}
@@ -27,4 +27,7 @@ testpaths = [
 ]
 
 [tool.pyright]
-include = ["validator"]
\ No newline at end of file
+include = ["validator"]
+
+[tool.setuptools.packages.find]
+include = ["validator", "inference", "validator.*", "inference.*"]
\ No newline at end of file