diff --git a/states/headscale/templates/update_policy.sh b/states/headscale/templates/update_policy.sh
index a9b783d..fbc0653 100644
--- a/states/headscale/templates/update_policy.sh
+++ b/states/headscale/templates/update_policy.sh
@@ -1,4 +1,8 @@
 #!/usr/bin/env zsh
 
 FILENAME=/etc/headscale/policy.json
-headscale policy get | jq > $FILENAME; vim $FILENAME; headscale policy set -f $FILENAME
+
+headscale policy get | jq > $FILENAME
+vim $FILENAME
+headscale policy set -f $FILENAME
+rm -f $FILENAME