diff --git a/roles/airgap/defaults/main.yml b/roles/airgap/defaults/main.yml
new file mode 100644
index 0000000..c28c766
--- /dev/null
+++ b/roles/airgap/defaults/main.yml
@@ -0,0 +1,3 @@
+---
+server_group: server # noqa var-naming[no-role-prefix]
+agent_group: agent # noqa var-naming[no-role-prefix]
diff --git a/roles/airgap/tasks/main.yml b/roles/airgap/tasks/main.yml
index 27d173e..3ed19de 100644
--- a/roles/airgap/tasks/main.yml
+++ b/roles/airgap/tasks/main.yml
@@ -100,7 +100,7 @@
     - name: Install Airgap K3s
       block:
         - name: Run K3s Install [server]
-          when: inventory_hostname in groups['server'] or ansible_host in groups['server']
+          when: inventory_hostname in groups[server_group] or ansible_host in groups[server_group]
           ansible.builtin.command:
             cmd: /usr/local/bin/k3s-install.sh
           environment:
@@ -109,7 +109,7 @@
           changed_when: true
 
         - name: Run K3s Install [agent]
-          when: inventory_hostname in groups['agent'] or ansible_host in groups['agent']
+          when: inventory_hostname in groups[agent_group] or ansible_host in groups[agent_group]
           ansible.builtin.command:
             cmd: /usr/local/bin/k3s-install.sh
           environment: