diff --git a/test/srt/test_two_batch_overlap.py b/test/srt/test_two_batch_overlap.py
index 02633b78a..108177e93 100644
--- a/test/srt/test_two_batch_overlap.py
+++ b/test/srt/test_two_batch_overlap.py
@@ -56,7 +56,7 @@ class TestTwoBatchOverlap(unittest.TestCase):
             },
         )
         print(f"{response.json()=}")
-        self.assertEquals(response.json()["text"], "5, 1+5=6")
+        self.assertEqual(response.json()["text"], "5, 1+5=6")
 
     def test_mmlu(self):
         args = SimpleNamespace(