diff --git a/META-INF/com/google/android/update-binary b/META-INF/com/google/android/update-binary
index bef5b53..e149a28 100755
--- a/META-INF/com/google/android/update-binary
+++ b/META-INF/com/google/android/update-binary
@@ -81,10 +81,10 @@ restore_env() {
 abort() {
   ui_print "$@";
   debugging;
+  restore_env;
   if [ ! -f anykernel.sh -o "$(file_getprop anykernel.sh do.cleanuponabort 2>/dev/null)" == 1 ]; then
     cleanup;
   fi;
-  restore_env;
   exit 1;
 }
 
@@ -280,11 +280,10 @@ if [ "$(file_getprop anykernel.sh do.modules)" == 1 ]; then
 fi;
 
 debugging;
+restore_env;
 
 if [ "$(file_getprop anykernel.sh do.cleanup)" == 1 ]; then
   cleanup;
 fi;
 
-restore_env;
-
 ui_print " " " " "Done!";