diff --git a/pyproject.toml b/pyproject.toml
index d2bed0e..1e353d7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,5 +1,5 @@
 [project]
-name = "smt"
+name = "smt-model"
 version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
@@ -19,3 +19,6 @@ dependencies = [
     "transformers>=4.49.0",
     "wandb>=0.19.7",
 ]
+
+[tool.setuptools]
+packages = ['vocab', 'config', 'graphics', 'smt_model', 'data_augmentation']