diff --git a/process_zip.sh b/process_zip.sh
index 8dc37c3..9de2408 100755
--- a/process_zip.sh
+++ b/process_zip.sh
@@ -63,7 +63,11 @@ function run() {
     TEST_PATH=$(realpath $TEST_PATH)
 
     mv $cipher/*.log "$TEST_PATH"
-    cp test_vectors/$VARIANT/LWC_AEAD_KAT_*.txt "$TEST_PATH/LWC_AEAD_KAT.txt" || exit 1
+
+    head -n 10 test_vectors/$VARIANT/LWC_AEAD_KAT_*.txt > "$TEST_PATH/LWC_AEAD_KAT.txt" || exit 1
+    echo "PT = 000102030405060708090A0B0C0D0E0F101112131415161718191A1B1C1D1E1F202122232425262728292A2B2C2D2E2F303132333435363738393A3B3C3D3E3F404142434445464748494A4B4C4D4E4F505152535455565758595A5B5C5D5E5F606162636465666768696A6B6C6D6E6F707172737475767778797A7B7C7D7E7F" >> $TEST_PATH/LWC_AEAD_KAT.txt
+    echo "AD = " >> $TEST_PATH/LWC_AEAD_KAT.txt
+
     mv $TMPDIR/$CIPHER_SLUG/*{.elf,.bin,.hex,.map} "$TEST_PATH"
 
     curl \
@@ -82,7 +86,7 @@ function run() {
 
   done
 
-  #rm -rf "$TMPDIR"
+  rm -rf "$TMPDIR"
 }
 
 if [[ $1 == "run" ]]; then
@@ -105,7 +109,7 @@ else
     touch $MAINDIR/locky.lock
     flock $MAINDIR/locky.lock $0 run $TMPDIR $TEMPLATE $MAINDIR/$TEMPLATE
   done
-  #rm -rf $TMPDIR
+  rm -rf $TMPDIR
 
 fi