diff --git a/.github/workflows/benchmarks.yml b/.github/workflows/benchmarks.yml
index c5c2bd72..6ceff42e 100644
--- a/.github/workflows/benchmarks.yml
+++ b/.github/workflows/benchmarks.yml
@@ -6,7 +6,7 @@ env:
 on:
   push:
     branches:
-      - "*"
+      - main
   pull_request_target:
     branches:
       - "*"
diff --git a/benchmarks/run_stardis.py b/benchmarks/run_stardis.py
index f70012c3..55ff3041 100644
--- a/benchmarks/run_stardis.py
+++ b/benchmarks/run_stardis.py
@@ -3,7 +3,6 @@
 import os
 import numpy as np
 from stardis.base import run_stardis
-
 from astropy import units as u
 
 from tardis.io.atom_data import AtomData
@@ -58,6 +57,7 @@ def setup(self):
         self.stellar_model = stellar_model
         self.alphas = alphas
         self.tracing_nus = tracing_nus
+        self.tracing_lambdas = tracing_lambdas
         self.config = config
 
     def time_run_stardis(self):
@@ -77,4 +77,4 @@ def time_calc_alpha(self):
             self.stellar_model,
             self.tracing_nus,
             self.config.opacity.line,
-        )
\ No newline at end of file
+        )