diff --git a/tests/embedding_net_test.py b/tests/embedding_net_test.py
index 09f59b177..eede90e4a 100644
--- a/tests/embedding_net_test.py
+++ b/tests/embedding_net_test.py
@@ -96,7 +96,7 @@ def test_iid_embedding_api(num_trials, num_dim):
 
 
 @pytest.mark.slow
-def test_iid_embedding_varying_num_trials(trial_factor=40, max_num_trials=100):
+def test_iid_embedding_varying_num_trials(trial_factor=50, max_num_trials=20):
     """Test embedding net with varying number of trials."""
     num_dim = 2
     prior = torch.distributions.MultivariateNormal(