diff --git a/wo/cli/templates/ufw.mustache b/wo/cli/templates/ufw.mustache
index aa8d344..8ff7150 100644
--- a/wo/cli/templates/ufw.mustache
+++ b/wo/cli/templates/ufw.mustache
@@ -59,7 +59,7 @@ wo_ufw_setup() {
 }
 
 if { wo_ufw_setup; }; then
-    exit 0
+    return 0
 else
-    exit 1
+    return 1
 fi