diff --git a/Jenkinsfile b/Jenkinsfile
index 12462e297c9f..1e55ce23499b 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -2545,6 +2545,7 @@ pipeline {
         trainer.accelerator=gpu \
         trainer.log_every_n_steps=1 \
         trainer.val_check_interval=10 \
+        +trainer.limit_val_batches=2 \
         trainer.accumulate_grad_batches=1 \
         trainer.max_steps=10 \
         trainer.precision=16 \
diff --git a/nemo/core/classes/modelPT.py b/nemo/core/classes/modelPT.py
index b84adb459eb9..c5cc99d5208d 100644
--- a/nemo/core/classes/modelPT.py
+++ b/nemo/core/classes/modelPT.py
@@ -15,7 +15,6 @@
 
 import copy
 import inspect
-import itertools
 import os
 import uuid
 from abc import abstractmethod