diff --git a/tests/tests_bec_ipython_client/test_x_ray_eye_align.py b/tests/tests_bec_ipython_client/test_x_ray_eye_align.py
index 8d83759..9a07aa3 100644
--- a/tests/tests_bec_ipython_client/test_x_ray_eye_align.py
+++ b/tests/tests_bec_ipython_client/test_x_ray_eye_align.py
@@ -2,8 +2,8 @@ from unittest import mock
 
 from bec_lib.device import DeviceBase
 
-from csaxs_bec.bec_ipython_client.plugins.LamNI import LamNI, XrayEyeAlign
-
+from csaxs_bec.bec_ipython_client.plugins.LamNI import LamNI
+from csaxs_bec.bec_ipython_client.plugins.LamNI.x_ray_eye_align import XrayEyeAlign
 # pylint: disable=unused-import