diff --git a/tests/proxy_unit_tests/test_db_schema_changes.py b/tests/proxy_unit_tests/test_db_schema_changes.py
index c7685763a24..5c6850c2f27 100644
--- a/tests/proxy_unit_tests/test_db_schema_changes.py
+++ b/tests/proxy_unit_tests/test_db_schema_changes.py
@@ -88,7 +88,7 @@ def test_aaaaaschema_compatibility():
     print("Current directory:", os.getcwd())
 
     # Get schemas
-    old_schema = get_schema_from_branch("main")
+    old_schema = get_schema_from_branch("litellm_stable_release_branch")
     with open("./schema.prisma", "r") as f:
         new_schema = f.read()