tests: fix recovery device tests due to additional screen

python/trezorlib/tests/device_tests/test_msg_recoverydevice_shamir.py

@@ -76,6 +76,11 @@ class TestMsgRecoveryDeviceShamir(TrezorTest):
                 self.client.debug.press_yes()
                 ret = self.client.call_raw(proto.ButtonAck())
 
+        # Confirm success
+        assert isinstance(ret, proto.ButtonRequest)
+        self.client.debug.press_yes()
+        ret = self.client.call_raw(proto.ButtonAck())
+
         # Workflow succesfully ended
         assert ret == proto.Success(message="Device recovered")
 
@@ -162,6 +167,11 @@ class TestMsgRecoveryDeviceShamir(TrezorTest):
         self.client.debug.input("654")
         ret = self.client.call_raw(proto.ButtonAck())
 
+        # Confirm success
+        assert isinstance(ret, proto.ButtonRequest)
+        self.client.debug.press_yes()
+        ret = self.client.call_raw(proto.ButtonAck())
+
         # Workflow succesfully ended
         assert ret == proto.Success(message="Device recovered")