diff --git a/pyproject.toml b/pyproject.toml
index 0254be4..75d24ca 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -50,4 +50,4 @@ app = [
 line-length = 120
 
 [tool.setuptools]
-packages = ["simple_einet"]
+packages = ["simple_einet", "simple_einet.layers", "simple_einet.layers.distributions"]