diff --git a/csaxs_bec/bec_ipython_client/plugins/LamNI/__init__.py b/csaxs_bec/bec_ipython_client/plugins/LamNI/__init__.py
index 286f52a..5642f93 100644
--- a/csaxs_bec/bec_ipython_client/plugins/LamNI/__init__.py
+++ b/csaxs_bec/bec_ipython_client/plugins/LamNI/__init__.py
@@ -1,2 +1,6 @@
-from .load_additional_correction import lamni_read_additional_correction
-from .x_ray_eye_align import DataDrivenLamNI, LamNI, MagLamNI, XrayEyeAlign
+from .alignment import XrayEyeAlign
+from .lamni import LamNI
+from .lamni_optics_mixin import LamNIInitError, LaMNIInitStages, LamNIOpticsMixin
+__all__ = [
+    "LamNI", "XrayEyeAlign", "LamNIInitError", "LaMNIInitStages", "LamNIOpticsMixin"
+]
\ No newline at end of file
diff --git a/csaxs_bec/bec_ipython_client/plugins/LamNI/lamni_optics_mixin (1).py b/csaxs_bec/bec_ipython_client/plugins/LamNI/lamni_optics_mixin.py
similarity index 100%
rename from csaxs_bec/bec_ipython_client/plugins/LamNI/lamni_optics_mixin (1).py
rename to csaxs_bec/bec_ipython_client/plugins/LamNI/lamni_optics_mixin.py