diff --git a/setup.py b/setup.py
index 1eb734e..09c37f1 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@ setup(
     package_dir={"": "src"},
     include_package_data=True,
     package_data={
-        "dotd": [
+        "drama_generator": [
             "saved_models/rnn_2epochs.pth",
             "data/shakespeare.txt",
             "data/markdown_header.txt",