diff --git a/check_home_assistant_state.py b/check_home_assistant_state.py
index 14a9c90..ca9d4c9 100755
--- a/check_home_assistant_state.py
+++ b/check_home_assistant_state.py
@@ -124,7 +124,14 @@ class Icinga2ConfAction(argparse.Action):
             if action.required:
                 print("      required = true")
 
-            if isinstance(action, argparse._StoreConstAction):
+            if isinstance(
+                action,
+                (
+                    argparse._StoreConstAction,
+                    argparse._CountAction,
+                    argparse._AppendConstAction,
+                ),
+            ):
                 print(f'      set_if = "${command_name}_{icinga2_var}$"')
             else:
                 print(f'      value = "${command_name}_{icinga2_var}$"')