diff --git a/sonoff-basic-r1.yaml b/sonoff-basic-r1.yaml
index 5ce776e..cce8449 100644
--- a/sonoff-basic-r1.yaml
+++ b/sonoff-basic-r1.yaml
@@ -115,10 +115,12 @@ sensor:
 
 text_sensor:
   - platform: wifi_info
-    entity_category: diagnostic
     ip_address:
       name: IP Address
+      entity_category: diagnostic
     ssid:
       name: Connected SSID
+      entity_category: diagnostic
     mac_address:
-      name: MAC Address
\ No newline at end of file
+      name: MAC Address
+      entity_category: diagnostic
\ No newline at end of file