diff --git a/lib/main.dart b/lib/main.dart
index e01bc25..8d3c33d 100644
--- a/lib/main.dart
+++ b/lib/main.dart
@@ -432,6 +432,7 @@ class _MainPageState extends State<MainPage> with WidgetsBindingObserver {
     Widget result;
     switch (entity.domain) {
       case "automation":
+      case "input_boolean":
       case "switch":
       case "light": {
         result = Switch(