diff --git a/falco/__init__.py b/falco/__init__.py
index 67ab82f..ca4d070 100644
--- a/falco/__init__.py
+++ b/falco/__init__.py
@@ -1,7 +1,11 @@
-from falco.config import init_from_mat as ifm
+from .config import init_from_mat as ifm  # ← CHANGED: Use relative import
+from . import config
+from . import model
+from . import proper
+from . import wfsc
+
 from .setup import *
 from .util import *
-# from .configs  import *
 from .imaging import *
 from .dm import *
 from .diff_dm import *
@@ -15,7 +19,6 @@
 from .hexsegmirror import *
 from .thinfilm import *
 from .zern import *
-from . import model
 from ._globals import INFLUENCE_XINETICS
 from ._globals import INFLUENCE_BMC_KILO
 from ._globals import INFLUENCE_BMC_2K