diff --git a/python/tests/test_jax.py b/python/tests/test_jax.py
index d20c8459f1..1447c38fb2 100644
--- a/python/tests/test_jax.py
+++ b/python/tests/test_jax.py
@@ -135,7 +135,7 @@ def _test_model(amici_module, jax_module, ts, p, k):
 
     amici_model.set_free_parameters(np.asarray(p, dtype=np.float64))
     amici_model.set_fixed_parameters(np.asarray(k, dtype=np.float64))
-    edata = ExpData(sol_amici_ref, 1.0, 1.0)
+    edata = ExpData(sol_amici_ref, 1.0, 1.0, 1)
     edata.free_parameters = amici_model.get_free_parameters()
     edata.fixed_parameters = amici_model.get_fixed_parameters()
     edata.pscale = amici_model.get_parameter_scale()
diff --git a/python/tests/test_sbml_import.py b/python/tests/test_sbml_import.py
index a8c1b951e8..774f0e9d78 100644
--- a/python/tests/test_sbml_import.py
+++ b/python/tests/test_sbml_import.py
@@ -558,10 +558,10 @@ def test_likelihoods(model_test_likelihoods):
 
     rdata = model.simulate(solver=solver)
     sigmas = rdata["y"].max(axis=0) * 0.05
-    edata = ExpData(rdata, sigmas, [])
+    edata = ExpData(rdata, sigmas, [], 1)
     # just make all observables positive since some are logarithmic
     while min(edata.get_measurements()) < 0:
-        edata = ExpData(rdata, sigmas, [])
+        edata = ExpData(rdata, sigmas, [], 1)
 
     # and now run for real and also compute likelihood values
     rdata = model.simulate(solver=solver, edata=[edata])[0]