diff --git a/test/models/transforms/test_input.py b/test/models/transforms/test_input.py
index 3200a22536..a9aa990d9e 100644
--- a/test/models/transforms/test_input.py
+++ b/test/models/transforms/test_input.py
@@ -218,7 +218,7 @@ def test_normalize(self):
 
                 nlz.eval()
                 X_unnlzd = nlz.untransform(X_nlzd)
-                self.assertTrue(torch.allclose(X, X_unnlzd, atol=1e-4, rtol=1e-4))
+                self.assertTrue(torch.allclose(X, X_unnlzd, atol=1e-3, rtol=1e-3))
                 expected_bounds = torch.cat(
                     [X.min(dim=-2, keepdim=True)[0], X.max(dim=-2, keepdim=True)[0]],
                     dim=-2,