diff --git a/base.yaml b/base.yaml
index fc6d0ba..b85db7e 100644
--- a/base.yaml
+++ b/base.yaml
@@ -225,8 +225,10 @@ light:
 
 button:
   - platform: restart
+    id: ${id_prefix}_restart
     name: "Restart"
   - platform: safe_mode
+    id: ${id_prefix}_safe_mode
     name: "Safe mode boot"
     entity_category: diagnostic
 
diff --git a/base_secplusv1.yaml b/base_secplusv1.yaml
index 18a61b6..be34dad 100644
--- a/base_secplusv1.yaml
+++ b/base_secplusv1.yaml
@@ -177,8 +177,10 @@ light:
 
 button:
   - platform: restart
+    id: ${id_prefix}_restart
     name: "Restart"
   - platform: safe_mode
+    id: ${id_prefix}_safe_mode
     name: "Safe mode boot"
     entity_category: diagnostic