diff --git a/src/ui/display.c b/src/ui/display.c
index b46be0eab..93ab6a486 100644
--- a/src/ui/display.c
+++ b/src/ui/display.c
@@ -76,12 +76,18 @@ static bool io_ui_process(dispatcher_context_t *context, bool set_dirty) {
     // We are not waiting for the client's input, nor we are doing computations on the device
     io_clear_processing_timeout();
 
+#ifdef REVAMPED_IO
+    do {
+        io_seproxyhal_io_heartbeat();
+    } while (!g_ux_flow_ended);
+#else   // !REVAMPED_IO
     io_seproxyhal_general_status();
     do {
         io_seproxyhal_spi_recv(G_io_seproxyhal_spi_buffer, sizeof(G_io_seproxyhal_spi_buffer), 0);
         io_seproxyhal_handle_event();
         io_seproxyhal_general_status();
     } while (io_seproxyhal_spi_is_status_sent() && !g_ux_flow_ended);
+#endif  // !REVAMPED_IO
 
     // We're back at work, we want to show the "Processing..." screen when appropriate
     io_start_processing_timeout();