diff --git a/f1_visualization/readme_machine.py b/f1_visualization/readme_machine.py
index 5ae739e..7129dac 100644
--- a/f1_visualization/readme_machine.py
+++ b/f1_visualization/readme_machine.py
@@ -148,7 +148,7 @@ def main(season: int, round_number: int, grand_prix: bool, update_readme: bool):
         .sort_values()
         .index
     )
-    team_palette = {team: p.team_color(team) for team in team_order}
+    team_palette = {team: p.get_team_color(team, session=session) for team in team_order}
 
     _, ax = plt.subplots(figsize=(15, 10))
     sns.boxplot(