diff --git a/test/mock/zenity b/test/mock/zenity
index 769de25..35270b4 100755
--- a/test/mock/zenity
+++ b/test/mock/zenity
@@ -2,20 +2,24 @@
 
 # fake-zenity: script that mocks the behavior of zenity based on the ./.fake-zenity-response file
 
+ZENITY_YES=0
+ZENITY_NO=1
+ZENITY_PERM=2
+
 if [[ $1 == "--set-fake-response" ]]; then
   #someone knows we are fake :)
-  echo $2 >~/.fake_zenity_response
+  echo "$2" >~/.fake_zenity_response
 else
   if [ -f ~/.fake_zenity_response ]; then
     FAKE_ZENITY_RESPONSE=$(cat ~/.fake_zenity_response)
 
+    printf "%s" "$4"
     if [[ $FAKE_ZENITY_RESPONSE == "yes_tmp" ]]; then
-      printf "Allow this time\n"
-      exit 1
+      exit "$ZENITY_YES"
     elif [[ $FAKE_ZENITY_RESPONSE == "no" ]]; then
-      exit 1
+      exit "$ZENITY_NO"
     elif [[ $FAKE_ZENITY_RESPONSE == "yes" ]]; then
-      exit 0
+      exit "$((ZENITY_YES | ZENITY_PERM))"
     fi
   fi
 fi