diff --git a/homeassistant/components/lifx/button.py b/homeassistant/components/lifx/button.py
index 3a4c73d2889..76afdc785e9 100644
--- a/homeassistant/components/lifx/button.py
+++ b/homeassistant/components/lifx/button.py
@@ -19,13 +19,13 @@ RESTART_BUTTON_DESCRIPTION = ButtonEntityDescription(
     key=RESTART,
     name="Restart",
     device_class=ButtonDeviceClass.RESTART,
-    entity_category=EntityCategory.DIAGNOSTIC,
+    entity_category=EntityCategory.CONFIG,
 )
 
 IDENTIFY_BUTTON_DESCRIPTION = ButtonEntityDescription(
     key=IDENTIFY,
     name="Identify",
-    entity_category=EntityCategory.DIAGNOSTIC,
+    entity_category=EntityCategory.CONFIG,
 )