diff --git a/app/src/display/widgets/output_status.c b/app/src/display/widgets/output_status.c
index 7e37f906..3bfbebf6 100644
--- a/app/src/display/widgets/output_status.c
+++ b/app/src/display/widgets/output_status.c
@@ -13,6 +13,7 @@ LOG_MODULE_DECLARE(zmk, CONFIG_ZMK_LOG_LEVEL);
 #include <zmk/event_manager.h>
 #include <zmk/events/usb_conn_state_changed.h>
 #include <zmk/events/ble_active_profile_changed.h>
+#include <zmk/events/endpoint_selection_changed.h>
 #include <zmk/usb.h>
 #include <zmk/ble.h>
 #include <zmk/endpoints.h>
@@ -86,6 +87,8 @@ int output_status_listener(const zmk_event_t *eh) {
 }
 
 ZMK_LISTENER(widget_output_status, output_status_listener)
+ZMK_SUBSCRIPTION(widget_output_status, zmk_endpoint_selection_changed);
+
 #if defined(CONFIG_USB)
 ZMK_SUBSCRIPTION(widget_output_status, zmk_usb_conn_state_changed);
 #endif