diff --git a/src/websocket.rs b/src/websocket.rs
index 5c9db73..c82bb0f 100644
--- a/src/websocket.rs
+++ b/src/websocket.rs
@@ -18,6 +18,7 @@ pub enum WsMessage {
     PointNewValue(crate::telemetry::PointMonitorInfo),
     PointSetValueBatchResult(crate::connection::BatchSetPointValueRes),
     EventCreated(crate::model::EventRecord),
+    UnitRuntimeChanged(crate::control::runtime::UnitRuntime),
 }
 
 #[derive(Debug, Clone, Serialize, Deserialize)]