diff --git a/self_compilation.sh b/self_compilation.sh
index e705ade..6806772 100644
--- a/self_compilation.sh
+++ b/self_compilation.sh
@@ -130,17 +130,15 @@ case "${distroId,,}" in
 
 
 # Remove sysmap/signing keys
-rm /lib/modules/"$KVER"/source/certs/signing_key*
-rm /lib/modules/"$KVER"/source/System.map
-rm /lib/modules/"$KVER"/source
-rm /lib/modules/"$KVER"/build
+rm /usr/src/linux-hardened-"$KVER"/certs/signing_key*
+rm /usr/src/linux-hardened-"$KVER"/System.map
+
 
 echo "Congrats! Your custom kernel based on the PlagueOS kernel configuration has been installed."
-echo "Reboot now? (y/N): "
-read reboot_opt
-
-if [[ "$reboot_opt" == "Y" ]]; then
-	reboot
+read -p "Reboot now? (y/N): " response
+response=$(echo $response | tr '[:upper:]' '[:lower:]')
+if [[ $response == "y" ]]; then
+    reboot
 else
-	exit
+    echo "Exiting..."
 fi
\ No newline at end of file