diff --git a/dkms-make.sh b/dkms-make.sh
index e24c191..780ced7 100755
--- a/dkms-make.sh
+++ b/dkms-make.sh
@@ -13,5 +13,6 @@ if [ "$sproc" -gt 1 ]; then
 	fi
 fi
 
-kernelver=${kernelver:-$(uname -r)}
-make "-j$sproc" "KVER=$kernelver" "KSRC=/lib/modules/$kernelver/build"
+KVER="${KVER:-$(uname -r)}"
+KSRC="/lib/modules/$KVER/build"
+make "-j$sproc" "KVER=$KVER" "KSRC=$KSRC"
diff --git a/install-driver.sh b/install-driver.sh
index 25966df..3c167b4 100755
--- a/install-driver.sh
+++ b/install-driver.sh
@@ -28,8 +28,9 @@ SCRIPT_VERSION="20230126"
 MODULE_NAME="88x2bu"
 DRV_VERSION="5.13.1"
 
-KARCH="$(uname -m)"
-KVER="$(uname -r)"
+KARCH="${KARCH:-$(uname -m)}"
+KVER="${KVER:-$(uname -r)}"
+KSRC="/lib/modules/$KVER/build"
 MODDESTDIR="/lib/modules/${KVER}/kernel/drivers/net/wireless/"
 
 DRV_NAME="rtl${MODULE_NAME}"
@@ -91,7 +92,7 @@ if ! command -v make >/dev/null 2>&1; then
 fi
 
 # check to see if the correct header files are installed
-if [ ! -d "/lib/modules/$(uname -r)/build" ]; then
+if [ ! -d "$KSRC" ]; then
 	echo "Your kernel header files aren't properly installed."
 	echo "Please consult your distro documentation or user support forums."
 	echo "Once the header files are properly installed, please run \"sudo ./${SCRIPT_NAME}\""
@@ -309,9 +310,9 @@ else
 	fi
 
 	if command -v /usr/bin/time >/dev/null 2>&1; then
-		/usr/bin/time -f "Compile time: %U seconds" dkms build -m ${DRV_NAME} -v ${DRV_VERSION}
+		/usr/bin/time -f "Compile time: %U seconds" dkms build -m ${DRV_NAME} -v ${DRV_VERSION} -k "$KVER" --kernelsourcedir "$KSRC"
 	else
-		dkms build -m ${DRV_NAME} -v ${DRV_VERSION}
+		dkms build -m ${DRV_NAME} -v ${DRV_VERSION} -k "$KVER" --kernelsourcedir "$KSRC"
 	fi
 	RESULT=$?
 
@@ -326,7 +327,7 @@ else
 		echo "The driver was built by dkms successfully."
 	fi
 
-	dkms install -m ${DRV_NAME} -v ${DRV_VERSION}
+	dkms install -m ${DRV_NAME} -v ${DRV_VERSION} -k "$KVER" --kernelsourcedir "$KSRC"
 	RESULT=$?
 
 	if [ "$RESULT" != "0" ]; then
diff --git a/remove-driver.sh b/remove-driver.sh
index ecd7182..9d97e27 100755
--- a/remove-driver.sh
+++ b/remove-driver.sh
@@ -28,8 +28,9 @@ SCRIPT_VERSION="20230126"
 MODULE_NAME="88x2bu"
 DRV_VERSION="5.13.1"
 
-KARCH="$(uname -m)"
-KVER="$(uname -r)"
+KARCH="${KARCH:-$(uname -m)}"
+KVER="${KVER:-$(uname -r)}"
+KSRC="/lib/modules/$KVER/build"
 MODDESTDIR="/lib/modules/${KVER}/kernel/drivers/net/wireless/"
 
 DRV_NAME="rtl${MODULE_NAME}"
@@ -104,7 +105,7 @@ fi
 if command -v dkms >/dev/null 2>&1; then
 	echo "Removing a dkms installation."
 	#  2>/dev/null suppresses the output of dkms
-	dkms remove -m ${DRV_NAME} -v ${DRV_VERSION} --all 2>/dev/null
+	dkms remove -m ${DRV_NAME} -v ${DRV_VERSION} -k "$KVER" --kernelsourcedir "$KSRC" --all 2>/dev/null
 	RESULT=$?
 	#echo "Result=${RESULT}"