diff --git a/tests/test_protection_levels.py b/tests/test_protection_levels.py
index bb550ac..14242de 100644
--- a/tests/test_protection_levels.py
+++ b/tests/test_protection_levels.py
@@ -150,7 +150,7 @@ class TestProtectionLevels(common.TrezorTest):
                                                 proto.PassphraseRequest(),
                                                 proto.ButtonRequest(code=proto_types.ButtonRequest_ConfirmOutput),
                                                 proto.ButtonRequest(code=proto_types.ButtonRequest_SignTx),
-                                                proto.TxRequest(finished=True)])
+                                                proto.TxRequest(request_type=proto_types.TXFINISHED)])
             self.client.simple_sign_tx('Bitcoin', [inp1, ], [out1, ])
 
     # def test_firmware_erase(self):
diff --git a/tests/test_zerosig.py b/tests/test_zerosig.py
index 15fa047..937d7ba 100644
--- a/tests/test_zerosig.py
+++ b/tests/test_zerosig.py
@@ -56,8 +56,8 @@ class TestZeroSig(common.TrezorTest):
                               )
 
         msg = self.client._prepare_simple_sign_tx('Bitcoin', [inp1], [out1])
-        tx = self.client.call(msg)   
-        siglen = ord(tx.serialized_tx[44])
+        tx = self.client.call(msg)
+        siglen = ord(tx.serialized.serialized_tx[44])
 
         # Trezor must strip leading zero from signature
         self.assertEqual(siglen, 67)
@@ -79,7 +79,7 @@ class TestZeroSig(common.TrezorTest):
 
         msg = self.client._prepare_simple_sign_tx('Bitcoin', [inp1], [out1])
         tx = self.client.call(msg)
-        siglen = ord(tx.serialized_tx[44])
+        siglen = ord(tx.serialized.serialized_tx[44])
 
         # Trezor must strip leading zero from signature
         self.assertEqual(siglen, 66)