diff --git a/module/action.sh b/module/action.sh
index 54d93fb..dfe7486 100644
--- a/module/action.sh
+++ b/module/action.sh
@@ -8,8 +8,8 @@ sh $MODPATH/autopif2.sh || exit 1
 
 echo -e "\nDone!"
 
-# warn since Magisk/KernelSU's implementation automatically closes if successful
-if [ "$APATCH" != "true" ]; then
+# warn since KernelSU's implementation automatically closes if successful
+if [ "$KSU" = "true" ]; then
     echo -e "\nClosing dialog in 20 seconds ..."
     sleep 20
 fi