diff --git a/pyproject.toml b/pyproject.toml
index 07dd972..fd6a6d9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -64,9 +64,6 @@ all = [
     "ipykernel",
 ]
 
-[project.entry-points."prototorch.plugins"]
-models = "prototorch.models"
-
 [build-system]
 requires = ["setuptools>=61", "wheel"]
 build-backend = "setuptools.build_meta"
@@ -91,6 +88,3 @@ line_length = 79
 [tool.mypy]
 explicit_package_bases = true
 namespace_packages = true
-
-[tool.setuptools]
-py-modules = ["prototorch"]
diff --git a/prototorch/models/__init__.py b/src/prototorch/models/__init__.py
similarity index 100%
rename from prototorch/models/__init__.py
rename to src/prototorch/models/__init__.py
diff --git a/prototorch/models/abstract.py b/src/prototorch/models/abstract.py
similarity index 100%
rename from prototorch/models/abstract.py
rename to src/prototorch/models/abstract.py
diff --git a/prototorch/models/callbacks.py b/src/prototorch/models/callbacks.py
similarity index 100%
rename from prototorch/models/callbacks.py
rename to src/prototorch/models/callbacks.py
diff --git a/prototorch/models/cbc.py b/src/prototorch/models/cbc.py
similarity index 100%
rename from prototorch/models/cbc.py
rename to src/prototorch/models/cbc.py
diff --git a/prototorch/models/extras.py b/src/prototorch/models/extras.py
similarity index 100%
rename from prototorch/models/extras.py
rename to src/prototorch/models/extras.py
diff --git a/prototorch/models/glvq.py b/src/prototorch/models/glvq.py
similarity index 100%
rename from prototorch/models/glvq.py
rename to src/prototorch/models/glvq.py
diff --git a/prototorch/models/knn.py b/src/prototorch/models/knn.py
similarity index 100%
rename from prototorch/models/knn.py
rename to src/prototorch/models/knn.py
diff --git a/prototorch/models/lvq.py b/src/prototorch/models/lvq.py
similarity index 100%
rename from prototorch/models/lvq.py
rename to src/prototorch/models/lvq.py
diff --git a/prototorch/models/probabilistic.py b/src/prototorch/models/probabilistic.py
similarity index 100%
rename from prototorch/models/probabilistic.py
rename to src/prototorch/models/probabilistic.py
diff --git a/prototorch/models/unsupervised.py b/src/prototorch/models/unsupervised.py
similarity index 100%
rename from prototorch/models/unsupervised.py
rename to src/prototorch/models/unsupervised.py
diff --git a/prototorch/models/vis.py b/src/prototorch/models/vis.py
similarity index 100%
rename from prototorch/models/vis.py
rename to src/prototorch/models/vis.py