diff --git a/gateway/include/devices.h b/gateway/include/devices.h
index fd186d3..342ab7d 100644
--- a/gateway/include/devices.h
+++ b/gateway/include/devices.h
@@ -114,13 +114,8 @@ struct EasyHomeSwitch : Switch {
 };
 
 Command* commands[] = {
-    Builder<Button>::instance(new Button{"Restart", "restart",
-        [](const char* msg) {
-            if (strcmp("PRESS", msg) == 0) ESP.restart();
-        }
-    })
+    HaESP::restartButton()
     .asDevice(gatewayDevice)
-    .withIcon("mdi:restart")
     .addPreconfigured(HaESP::heapStats)
     .addPreconfigured(HaESP::restartInfo)
     .build(),