diff --git a/benchmark/over_time.py b/benchmark/over_time.py
index 455a07ccb..98d206794 100644
--- a/benchmark/over_time.py
+++ b/benchmark/over_time.py
@@ -12,7 +12,7 @@ def get_model_color(model):
     if model == "gpt-4o-mini":
         return default
 
-    if "claude-3-sonnet" in model.lower():
+    if "sonnet" in model.lower():
         return "orange"
 
     if "-4o" in model: