diff --git a/backend/setup.py b/backend/setup.py
index 8bc9b6d4..1f836765 100644
--- a/backend/setup.py
+++ b/backend/setup.py
@@ -48,7 +48,7 @@
         'ipykernel >= 5.1.4',
         'notebook >= 6.0.0',
         'packaging > 20',
-        'ml_metadata >= 0.26.0, < 1',
+        'ml_metadata >= 1.0.0, < 2',
         'progress >= 1.5',
         'kfserving >= 0.4.0, < 0.5.0',
         'kubernetes < 12.0.0',