diff --git a/tests/device_tests/test_msg_wipedevice.py b/tests/device_tests/test_msg_wipedevice.py
index 972e6186c6..8275bfc715 100644
--- a/tests/device_tests/test_msg_wipedevice.py
+++ b/tests/device_tests/test_msg_wipedevice.py
@@ -43,13 +43,13 @@ def test_wipe_device(client: Client):
 
 
 @pytest.mark.setup_client(pin=PIN4)
-def test_autolock_not_retained(client: Client):
-    session = client.get_session()
+def test_autolock_not_retained(session: Session):
+    client = session.client
     with client:
         client.use_pin_sequence([PIN4])
         device.apply_settings(session, auto_lock_delay_ms=10_000)
 
-    assert client.features.auto_lock_delay_ms == 10_000
+    assert session.features.auto_lock_delay_ms == 10_000
 
     device.wipe(session)
     client = client.get_new_client()