diff --git a/test_extract.sh b/test_extract.sh
index 8c1a2df..8920cb4 100755
--- a/test_extract.sh
+++ b/test_extract.sh
@@ -6,6 +6,7 @@ rm -rf tmp
 mkdir tmp
 
 for i in tests/*; do
+    i=$(basename "$i")
     echo "Testing $i"
     mkdir -p "tmp/$i/a" "tmp/$i/b"
     ./aastuff "tests/$i/encrypted.aar" "tmp/$i/a" "$(cat tests/"$i"/key.txt)"
diff --git a/test_get_key.sh b/test_get_key.sh
index bc698fa..1b08d33 100755
--- a/test_get_key.sh
+++ b/test_get_key.sh
@@ -11,6 +11,7 @@ rm -rf tmp
 mkdir tmp
 
 for i in tests/*; do
+    i=$(basename "$i")
     echo "Testing $i"
     mkdir -p "tmp/$i"
     # Ensure expected key is valid first