diff --git a/pyproject.toml b/pyproject.toml
index 4baba1f..606c489 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,6 +25,9 @@ dependencies = [
 "repository" = "https://git.weber.codes/tom/drama_generator.git"
 
 
+[tool.setuptools.packages.find]
+where = ["src"]
+
 [tool.setuptools.package-data]
 drama_generator.data = ["shakespeare.txt"]
 drama_generator.saved_models = ["rnn_2epochs.pth"]