diff --git a/tensorboard_plugin/setup.py b/tensorboard_plugin/setup.py
index d09e844..2c1c50b 100644
--- a/tensorboard_plugin/setup.py
+++ b/tensorboard_plugin/setup.py
@@ -47,7 +47,7 @@ def select_constraint(default, nightly=None, git_master=None):
     'tensorflow>=2.15,<3',
     'tensorflow-model-analysis'
     + select_constraint(
-        default='>=0.45,<0.46',
+        default='>=0.45,<=0.46',
         nightly='>=0.46.0.dev',
         git_master='@git+https://github.com/tensorflow/model-analysis@master',
     ),