diff --git a/base.yaml b/base.yaml
index 4ba4068fd74b89d93686c7cc4b3d55a5a7acc844..0dec1a5b65ac28c0fc8c1b30a6c9f15766926aec 100644
--- a/base.yaml
+++ b/base.yaml
@@ -73,6 +73,6 @@ text_sensor:
     bssid:
       name: "Wifi BSSID"
 
-switch:
+button:
   - platform: restart
     name: "Device Restart"