diff --git a/CommonInstaller b/CommonInstaller
index 85e5166..5d7fed8 100644
--- a/CommonInstaller
+++ b/CommonInstaller
@@ -176,7 +176,8 @@ error () {
 
 	if [ "${MODE}" = "MAGISK" ]; then
 		rm -rf "${MODPATH}"
-		magisk_install_cleanup
+		unmount_magisk_img
+		${BOOTMODE} || recovery_cleanup
 	fi
 
 	nanodroid_storelogs