diff --git a/pyproject.toml b/pyproject.toml
index 9234c93..4baba1f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,5 +25,6 @@ dependencies = [
 "repository" = "https://git.weber.codes/tom/drama_generator.git"
 
 
-[tool.setuptools.packages.find]
-where = ["src"]
\ No newline at end of file
+[tool.setuptools.package-data]
+drama_generator.data = ["shakespeare.txt"]
+drama_generator.saved_models = ["rnn_2epochs.pth"]