diff --git a/test/mx/distribution/test_mx_distribution_inference.py b/test/mx/distribution/test_mx_distribution_inference.py
index b26a2d2ce7..753a084cf0 100644
--- a/test/mx/distribution/test_mx_distribution_inference.py
+++ b/test/mx/distribution/test_mx_distribution_inference.py
@@ -499,6 +499,7 @@ def test_dirichlet(hybridize: bool) -> None:
     ), f"Covariance did not match: cov = {cov}, cov_hat = {cov_hat}"
 
 
+@pytest.mark.flaky(retries=3)
 @pytest.mark.parametrize("hybridize", [True, False])
 def test_dirichlet_multinomial(hybridize: bool) -> None:
     num_samples = 2000