diff --git a/include/devices.h b/include/devices.h
index 56e114e..c25c33f 100644
--- a/include/devices.h
+++ b/include/devices.h
@@ -82,12 +82,7 @@ namespace Devices {
         }).withIcon("mdi:timer-play-outline").restoreStateFromCommand().build();
 
     auto button = 
-        Builder<Button>::instance(new Button{"Restart", "restart",
-            [](const char* msg) {
-                if (strcmp("PRESS", msg) == 0) ESP.restart();
-            }
-        })
-        .withIcon("mdi:restart")
+        HaESP::restartButton()
         .asDevice(espClockDevice)
         .addSecondary(
             Builder<Button>::instance(new Button{"Display time", "display_time",