fixup! test: update device tests [no changelog]

trezor · Mar 6, 2025 · 7bd29f8 · 7bd29f8
1 parent 1ee6537
commit 7bd29f8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/device_tests/test_session.py b/tests/device_tests/test_session.py
@@ -76,7 +76,7 @@ def test_end_session(client: Client):
     assert session.id is not None
 
     # get_address will succeed
-    with session as session:
+    with session:
         session.set_expected_responses([messages.Address])
         get_test_address(session)
 
@@ -89,7 +89,7 @@ def test_end_session(client: Client):
 
     session = client.get_session()
     assert session.id is not None
-    with session as session:
+    with session:
         session.set_expected_responses([messages.Address])
         get_test_address(session)