diff --git a/templates/mystrom.yaml b/templates/mystrom.yaml
index 02b0986..c27a822 100644
--- a/templates/mystrom.yaml
+++ b/templates/mystrom.yaml
@@ -120,3 +120,4 @@ switch:
   pin: GPIO27
   id: mystrom_relay
   restore_mode: RESTORE_DEFAULT_ON
+  device_class: outlet