diff --git a/dune-mpf/__init__.py b/dune-mpf/__init__.py
new file mode 100644
index 0000000..023cf03
--- /dev/null
+++ b/dune-mpf/__init__.py
@@ -0,0 +1,5 @@
+from .dune_mpf import *
+
+__doc__ = dune_mpf.__doc__
+if hasattr(dune_mpf, "__all__"):
+    __all__ = dune_mpf.__all__
diff --git a/dune-mpf/dune_mpf/__init__.py b/dune-mpf/dune_mpf/__init__.py
index f9146e4..aef921f 100644
--- a/dune-mpf/dune_mpf/__init__.py
+++ b/dune-mpf/dune_mpf/__init__.py
@@ -4,7 +4,7 @@
 import pytoml
 import yaml
 
-import dune_mpf
+from 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):
-        import dune_mpf
+        from dune_mpf import dune_mpf
 
     # Send DUNE context to every phynode
     mpf.client[:].push({'dune': dune})
diff --git a/dune-mpf/pyproject.toml b/dune-mpf/pyproject.toml
index 91ffad7..0d58ef5 100644
--- a/dune-mpf/pyproject.toml
+++ b/dune-mpf/pyproject.toml
@@ -1,11 +1,11 @@
+[project]
+name = "dune"
+dependencies = ["pytoml", "pyyaml", "mpf @ git+https://github.com/mpiraux/mpf.git"]
+
 [build-system]
 requires = ["maturin>=1.0,<2.0"]
 build-backend = "maturin"
 
 [tool.maturin]
-# "extension-module" tells pyo3 we want to build an extension module (skips linking against libpython.so)
 features = ["pyo3/extension-module"]
-# target = ["aarch64-unknown-linux-gnu", "x86_64-unknown-linux-gnu"]
-# [tool.maturin.target]
-# arch = "aarch64"
-
+name = "dune"