diff --git a/tests/device_handler.py b/tests/device_handler.py
index 9edf0b560d..a3dc61314d 100644
--- a/tests/device_handler.py
+++ b/tests/device_handler.py
@@ -53,6 +53,7 @@ class BackgroundDeviceHandler:
     def _configure_client(self, client: "Client") -> None:
         self.client = client
         self.client.ui = NullUI  # type: ignore [NullUI is OK UI]
+        self.client.button_callback = self.client.ui.button_request
         self.client.watch_layout(True)
 
     def run_with_session(