diff --git a/scripts/protein/esm2/test_esm2_pretrain.py b/scripts/protein/esm2/test_esm2_pretrain.py
index 9bbbde489..8e4303b32 100644
--- a/scripts/protein/esm2/test_esm2_pretrain.py
+++ b/scripts/protein/esm2/test_esm2_pretrain.py
@@ -106,8 +106,8 @@ def test_main_runs(monkeypatch, tmpdir, dummy_protein_dataset, dummy_parquet_tra
             wandb_offline=True,
             num_steps=55,
             warmup_steps=5,
-            limit_val_batches=5,
-            val_check_interval=10,
+            limit_val_batches=1,
+            val_check_interval=1,
             log_every_n_steps=None,
             num_dataset_workers=1,
             biobert_spec_option=BiobertSpecOption.esm2_bert_layer_with_transformer_engine_spec,