diff --git a/trezorlib/tests/device_tests/test_protection_levels.py b/trezorlib/tests/device_tests/test_protection_levels.py
index 729ae0a069..7f1955b758 100644
--- a/trezorlib/tests/device_tests/test_protection_levels.py
+++ b/trezorlib/tests/device_tests/test_protection_levels.py
@@ -134,7 +134,8 @@ class TestProtectionLevels(TrezorTest):
     def test_reset_device(self):
         with self.client:
             self.client.set_expected_responses(
-                [proto.EntropyRequest()]
+                [proto.ButtonRequest()]
+                + [proto.EntropyRequest()]
                 + [proto.ButtonRequest()] * 24
                 + [proto.Success(), proto.Features()]
             )