diff --git a/tests/e2e/test_relora_llama.py b/tests/e2e/test_relora_llama.py
index 70310923d..5de5db11b 100644
--- a/tests/e2e/test_relora_llama.py
+++ b/tests/e2e/test_relora_llama.py
@@ -58,7 +58,6 @@ class TestReLoraLlama(unittest.TestCase):
                 "output_dir": temp_dir,
                 "learning_rate": 0.00001,
                 "optimizer": "adamw_torch",
-                "max_steps": 5,
                 "lr_scheduler": "cosine",
             }
         )