diff --git a/src/run.py b/src/run.py
index eda0373c1e609e0818e98358d00a00beddb63cdf..11029817a978b872d0f99954a50ab5f5b93aa012 100644
--- a/src/run.py
+++ b/src/run.py
@@ -26,7 +26,8 @@ def run(stations=['DEBW107', 'DEBY081', 'DEBW013', 'DEBW076', 'DEBW087', 'DEBW00
         evaluate_bootstraps=True, number_of_bootstraps=None, create_new_bootstraps=False,
         plot_list=None,
         model=None,
-        batch_size=None):
+        batch_size=None,
+        epochs=None):
 
     params = inspect.getfullargspec(ExperimentSetup).args
     kwargs = {k: v for k, v in locals().items() if k in params}