diff --git a/src/main/python/calibrate.py b/src/main/python/calibrate.py
index bb3f99b..3ffcdf0 100644
--- a/src/main/python/calibrate.py
+++ b/src/main/python/calibrate.py
@@ -63,7 +63,7 @@ def filter_modes(df):
 
 study, obj = create_calibration(
     "calib",
-    ASCCalibrator(modes, initial, target, lr=utils.linear_lr_scheduler(start=0.3, interval=12)),
+    ASCCalibrator(modes, initial, target, lr=utils.linear_scheduler(start=0.3, interval=12)),
     "matsim-lausitz-1.x-SNAPSHOT-20c8ab3.jar",
     "../input/v1.0/lausitz-v1.0-25pct.config.xml",
     args="--25pct",
diff --git a/src/main/sh/setup.sh b/src/main/sh/setup.sh
index dd0a975..25d79fa 100644
--- a/src/main/sh/setup.sh
+++ b/src/main/sh/setup.sh
@@ -5,7 +5,7 @@ source env/bin/activate
 
 pip install --upgrade pip
 
-pip install "matsim-tools[calibration]==0.0.16"
+pip install "matsim-tools[calibration]==0.0.18"
 
 # Alternative: Install from a specific branch on github
 #pip install --force-reinstall "matsim-tools[calibration] @ git+https://github.com/matsim-vsp/matsim-python-tools.git@dist-calibration"
\ No newline at end of file