diff --git a/profit_simulation.py b/profit_simulation.py
index 446a217..1febfc3 100644
--- a/profit_simulation.py
+++ b/profit_simulation.py
@@ -97,7 +97,7 @@ def plot_won_lost_mean_odds(data_df: pd.DataFrame):
     data_df = data_df.sort_values(by='date')
     date_x = data_df['date'].tolist()
     fig = go.Figure()
-    cols = data_df.columns[1:]
+    cols = ['won', 'lost']
     for col in cols:
         y_data = data_df[col].tolist()
         fig.add_trace(go.Bar(x=date_x, y=y_data, name=col, yaxis='y1'))