diff --git a/tests/ui_tests/__init__.py b/tests/ui_tests/__init__.py
index 1fe82a98a5..5ae6a89c82 100644
--- a/tests/ui_tests/__init__.py
+++ b/tests/ui_tests/__init__.py
@@ -60,7 +60,7 @@ def screen_recording(
     finally:
         if client_callback:
             client = client_callback()
-        if client.protocol_version == ProtocolVersion.PROTOCOL_V1:
+        if client.protocol_version == ProtocolVersion.V1:
             client.sync_responses()
         # Wait for response to Initialize, which gives the emulator time to catch up
         # and redraw the homescreen. Otherwise there's a race condition between that