diff --git a/ima-setup.sh b/ima-setup.sh
index 3fa8b34..4a61a28 100755
--- a/ima-setup.sh
+++ b/ima-setup.sh
@@ -33,7 +33,7 @@ for _opt in "$@"; do
 	--policy=*)
 		ima_policy_path=${_opt#*=}
 		if [[ ! -e $ima_policy_path ]]; then
-			echo "$policy_file doesn't exist"
+			echo "$ima_policy_path doesn't exist"
 			exit 1
 		fi
 		;;
@@ -61,7 +61,7 @@ if test -f /run/ostree-booted; then
 	echo "You are using OSTree, please enable IMA signatures as part of the OSTree creation process."
 else
 	echo "Adding IMA signatures to installed package files"
-	if ! ima-add-sigs; then
+	if ! ima-add-sigs --reinstall_threshold="$reinstall_threshold"; then
 		echo "Failed to add IMA signatures, abort"
 		exit 1
 	fi