diff --git a/tests/algorithms/test_algorithms_train.py b/tests/algorithms/test_algorithms_train.py
index 53d871265c..0f9cb32de2 100644
--- a/tests/algorithms/test_algorithms_train.py
+++ b/tests/algorithms/test_algorithms_train.py
@@ -18,7 +18,7 @@ def test_algorithm_trains(alg_cls: type[Algorithm]):
     trainer = Trainer(
         model=model,
         train_dataloader=dataloader,
-        max_duration='1ba',
+        max_duration='2ba',
         algorithms=alg_cls(**alg_kwargs),
     )
     trainer.fit()
@@ -34,5 +34,5 @@ def test_algorithm_trains(alg_cls: type[Algorithm]):
             'GyroDropout is implemented to be applied on Event.FIT_START, so is not compatible with multiple calls to fit.',
         )
 
-    # fit again for another epoch
+    # fit again for another batch
     trainer.fit(duration='1ba')