diff --git a/tests/functional/test_cases/e2e-nemo2 b/tests/functional/test_cases/e2e-nemo2
index d56d7bae6..a51a0360e 100755
--- a/tests/functional/test_cases/e2e-nemo2
+++ b/tests/functional/test_cases/e2e-nemo2
@@ -44,9 +44,9 @@ python /opt/NeMo/tests/collections/llm/megatron_gpt_pretraining.py \
   --devices=1 \
   --max-steps=3 \
   --experiment-dir=${PRETRAINING_PATH} \
-  --vocab-path=/datadrive/TestData/nlp/megatron_gpt/data/gpt/vocab.json \
-  --merges-path=/datadrive/TestData/nlp/megatron_gpt/data/gpt/merges.txt \
-  --data-path=/datadrive/TestData/nlp/megatron_gpt/data/gpt/simple_wiki_gpt_preproc_text_document \
+  --vocab-path=${ALIGNER_CI_DIR}/nlp-copy/megatron_gpt/data/gpt/vocab.json \
+  --merges-path=${ALIGNER_CI_DIR}/nlp-copy/megatron_gpt/data/gpt/merges.txt \
+  --data-path=${ALIGNER_CI_DIR}/nlp-copy/megatron_gpt/data/gpt/simple_wiki_gpt_preproc_text_document \
   --index-mapping-dir=tests/collections/llm/gpt_index_mappings \
   --no-masked-softmax-fusion