diff --git a/tests/test_galil.py b/tests/test_galil.py
index f4814a2..ba98480 100644
--- a/tests/test_galil.py
+++ b/tests/test_galil.py
@@ -21,7 +21,7 @@ def test_axis_get(pos, msg, sign):
         socket_cls=SocketMock,
     )
     leyey.controller.sock.flush_buffer()
-    leyey.stage()
+    leyey.controller.on()
     leyey.controller.sock.buffer_recv = msg
     val = leyey.read()
     assert val["leyey"]["value"] == pos
diff --git a/tests/test_smaract.py b/tests/test_smaract.py
index af3f624..6ad5588 100644
--- a/tests/test_smaract.py
+++ b/tests/test_smaract.py
@@ -203,7 +203,7 @@ def test_move_axis(pos, get_msg, return_msg):
         sign=1,
         socket_cls=SocketMock,
     )
-    lsmarA.stage()
+    lsmarA.controller.on()
     controller = lsmarA.controller
     controller.sock.flush_buffer()
     controller.sock.buffer_recv = return_msg