diff --git a/fastvideo/tests/training/VSA/test_training_loss_VSA.py b/fastvideo/tests/training/VSA/test_training_loss_VSA.py
index 6b69ecd24..b4dd1fc95 100644
--- a/fastvideo/tests/training/VSA/test_training_loss_VSA.py
+++ b/fastvideo/tests/training/VSA/test_training_loss_VSA.py
@@ -121,9 +121,9 @@ def test_distributed_training():
     wandb_summary = json.load(open(summary_file))
 
     fields_and_thresholds = {
-        'avg_step_time': 3,
-        'grad_norm': 0.2,
-        'step_time': 2.5,
+        'avg_step_time': 5,
+        'grad_norm': 0.5,
+        'step_time': 5,
         'train_loss': 0.04
     }