diff --git a/dune-mpf/pyproject.toml b/dune-mpf/pyproject.toml
index 0d58ef5..58501c7 100644
--- a/dune-mpf/pyproject.toml
+++ b/dune-mpf/pyproject.toml
@@ -8,4 +8,5 @@ build-backend = "maturin"
 
 [tool.maturin]
 features = ["pyo3/extension-module"]
+python-source = "python"
 name = "dune"
diff --git a/dune-mpf/dune_mpf/__init__.py b/dune-mpf/python/dune/__init__.py
similarity index 94%
rename from dune-mpf/dune_mpf/__init__.py
rename to dune-mpf/python/dune/__init__.py
index aef921f..f9146e4 100644
--- a/dune-mpf/dune_mpf/__init__.py
+++ b/dune-mpf/python/dune/__init__.py
@@ -4,7 +4,7 @@
 import pytoml
 import yaml
 
-from dune_mpf import dune_mpf
+import dune_mpf
 import mpf
 
 
@@ -34,7 +34,7 @@ def setup():
 
     # Import DUNE module on every phynode
     with mpf.client[:].sync_imports(quiet=False):
-        from dune_mpf import dune_mpf
+        import dune_mpf
 
     # Send DUNE context to every phynode
     mpf.client[:].push({'dune': dune})