diff --git a/tests/collections/llm/test_mnist_model_nemo2.py b/tests/collections/llm/test_mnist_model_nemo2.py
index 96e4aefbf228..a09ed503aa89 100644
--- a/tests/collections/llm/test_mnist_model_nemo2.py
+++ b/tests/collections/llm/test_mnist_model_nemo2.py
@@ -569,6 +569,7 @@ def test_train_mnist_litautoencoder_with_megatron_strategy_single_gpu(tmpdir):
             accelerator="gpu",
             devices=1,
             strategy=pred_strategy,
+            default_root_dir=root_dir,
         )
         ckpt_path = checkpoint_callback.last_model_path.replace(
             ".ckpt", ""