diff --git a/requirements/optional.txt b/requirements/optional.txt
index 85853cda452..5f31808f14b 100644
--- a/requirements/optional.txt
+++ b/requirements/optional.txt
@@ -1,4 +1,4 @@
 albumentations>=0.3.2 --no-binary qudida,albumentations    # For Albumentations data transform
-grad-cam >= 1.3.7   # For CAM visualization
+grad-cam >= 1.3.7,<1.5.0   # For CAM visualization
 requests            # For torchserve
 scikit-learn        # For t-SNE visualization and unit tests.
diff --git a/tests/test_models/test_backbones/test_repmlp.py b/tests/test_models/test_backbones/test_repmlp.py
index bfcb5dfccb1..f03fce4ed41 100644
--- a/tests/test_models/test_backbones/test_repmlp.py
+++ b/tests/test_models/test_backbones/test_repmlp.py
@@ -169,4 +169,5 @@ def test_deploy_(self):
 
         assert len(feats_) == len(feats__)
         for i in range(len(feats)):
-            self.assertTrue(torch.allclose(feats__[i], feats_[i]))
+            self.assertTrue(
+                torch.allclose(feats__[i], feats_[i], rtol=0.01, atol=0.01))