a

2 files changed, 3 insertions, 1 deletions

diff --git a/slackware-bootstrap/step2 b/slackware-bootstrap/step2
index a7f25ec..9ac5f3f 100755
--- a/slackware-bootstrap/step2
+++ b/slackware-bootstrap/step2
@@ -18,6 +18,7 @@ if [[ ! -f "~/.multilib-done" ]]; then
   sh $basedir/multilib
 fi
 
+nreboot=0
 if [[ ! -f "~/.kernel-done" ]]; then
   echo "============== [ kernel update ] ==============="
   echo "you can choose to update your kernel"
@@ -25,7 +26,6 @@ if [[ ! -f "~/.kernel-done" ]]; then
   echo "this will update your kernel to $kernelver"
   echo "================================================"
 
-  nreboot=0
 
   read -p "do you want to update your kernel [y/n]? (y):" choice
   if [[ $choice == "y" ]]; then
diff --git a/slackware-bootstrap/step2-finish b/slackware-bootstrap/step2-finish
index 595f5a3..da58e62 100755
--- a/slackware-bootstrap/step2-finish
+++ b/slackware-bootstrap/step2-finish
@@ -1,5 +1,7 @@
 #!/bin/bash
 
+basedir=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
+
 if [[ -f "~/.nreboot" ]]; then
   echo "sh $basedir/finish" >> /root/.bash_profile
   echo "rm -f /root/.bash_profile" >> /root/.bash_profile