diff --git a/lib/entity_class/entity.class.dart b/lib/entity_class/entity.class.dart
index 7908b96..aa168a1 100644
--- a/lib/entity_class/entity.class.dart
+++ b/lib/entity_class/entity.class.dart
@@ -106,7 +106,7 @@ class EntityWidget extends StatefulWidget {
   _EntityWidgetState createState() {
     switch (entity.domain) {
       case "automation":
-      case "input_boolean ":
+      case "input_boolean":
       case "switch":
       case "light": {
         return _SwitchEntityWidgetState();