diff --git a/end_to_end/tpu/gemma2/2b/test_gemma2_to_mt.sh b/end_to_end/tpu/gemma2/2b/test_gemma2_to_mt.sh
index 7daba89af..ac0a504c8 100644
--- a/end_to_end/tpu/gemma2/2b/test_gemma2_to_mt.sh
+++ b/end_to_end/tpu/gemma2/2b/test_gemma2_to_mt.sh
@@ -64,7 +64,7 @@ python3 -m MaxText.decode "${MAXTEXT_PKG_DIR:-${MAXTEXT_REPO_ROOT:-$PWD}/src/Max
 # Non-Googlers please remember to point `DATASET_PATH` to the GCS bucket where you have your training data
 export DATASET_PATH=gs://maxtext-dataset
 # Non-Googlers please remember to point `BASE_OUTPUT_DIRECTORY` to a GCS bucket that you own, this bucket will store all the files generated by MaxText during a run
-export BASE_OUTPUT_DIRECTORY=gs://runner-maxtext-logs
+export BASE_OUTPUT_DIRECTORY=gs://runner-maxtext-logs/gemm2-2b
 
 # We can also run finetuning by using the scanned converted checkpoint.
 # Note that scanned checkpoint helps with efficient finetuning
diff --git a/end_to_end/tpu/gemma3/4b/test_gemma3_to_mt.sh b/end_to_end/tpu/gemma3/4b/test_gemma3_to_mt.sh
index f40951564..4b48b0307 100644
--- a/end_to_end/tpu/gemma3/4b/test_gemma3_to_mt.sh
+++ b/end_to_end/tpu/gemma3/4b/test_gemma3_to_mt.sh
@@ -75,7 +75,7 @@ fi
 # Non-Googlers please remember to point `DATASET_PATH` to the GCS bucket where you have your training data
 export DATASET_PATH=gs://maxtext-dataset
 # Non-Googlers please remember to point `BASE_OUTPUT_DIRECTORY` to a GCS bucket that you own, this bucket will store all the files generated by MaxText during a run
-export BASE_OUTPUT_DIRECTORY=gs://runner-maxtext-logs
+export BASE_OUTPUT_DIRECTORY=gs://runner-maxtext-logs/gemma3-4b
 
 # We can also run finetuning by using the scanned converted checkpoint.
 # Note that scanned checkpoint helps with efficient finetuning