diff --git a/src/main.rs b/src/main.rs
index f1a3a82..36aa5ff 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -217,6 +217,22 @@ fn build_router(state: AppState) -> Router {
             "/api/control/equipment/{equipment_id}/stop",
             post(handler::control::stop_equipment),
         )
+        .route(
+            "/api/control/unit/{unit_id}/start-auto",
+            post(handler::control::start_auto_unit),
+        )
+        .route(
+            "/api/control/unit/{unit_id}/stop-auto",
+            post(handler::control::stop_auto_unit),
+        )
+        .route(
+            "/api/control/unit/{unit_id}/ack-fault",
+            post(handler::control::ack_fault_unit),
+        )
+        .route(
+            "/api/unit/{unit_id}/runtime",
+            get(handler::control::get_unit_runtime),
+        )
         .route(
             "/api/tag",
             get(handler::tag::get_tag_list).post(handler::tag::create_tag),