diff --git a/terraform/modules/k8s-cluster/templates/cloud-init-k8s.tftpl b/terraform/modules/k8s-cluster/templates/cloud-init-k8s.tftpl
index 9e92183..a8f0394 100644
--- a/terraform/modules/k8s-cluster/templates/cloud-init-k8s.tftpl
+++ b/terraform/modules/k8s-cluster/templates/cloud-init-k8s.tftpl
@@ -12,13 +12,15 @@ users:
     ssh_authorized_keys:
       - ${terraform_public_ssh_key}
 packages:
-  - fail2ban
-  - ufw
+  - apt:
+    - snapd
+    - fail2ban
+    - ufw
+  - snap:
+    - [kubectl, --classic]
 package_update: true
 package_upgrade: true
-snap:
-  commands:
-  - snap install kubectl --classic
+package_reboot_if_required: true
 write_files:
   - owner: root:root
     content: "${microk8s_config}"