diff --git a/pyproject.toml b/pyproject.toml
index 1652ff4..a790f32 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -69,9 +69,8 @@ classifiers = [
 ]
 packages = [
     {include = "notifications/*.py"},
-    {include = "notifications/base/"},
     {include = "notifications/migrations/"},
-    {include = "notifications/templatetags/"},
+    {include = "notifications/models/"},
     {include = "notifications/templatetags/"},
 ]
 include = [