diff --git a/python/sdist/amici/petab/cli/import_petab.py b/python/sdist/amici/petab/cli/import_petab.py
index db600b0590..9f29d26e71 100644
--- a/python/sdist/amici/petab/cli/import_petab.py
+++ b/python/sdist/amici/petab/cli/import_petab.py
@@ -145,10 +145,7 @@ def _main():
 
     import_model_sbml(
         model_name=args.model_name,
-        sbml_model=pp.sbml_model,
-        condition_table=pp.condition_df,
-        observable_table=pp.observable_df,
-        measurement_table=pp.measurement_df,
+        petab_problem=pp,
         model_output_dir=args.model_output_dir,
         compile=args.compile,
         generate_sensitivity_code=args.generate_sensitivity_code,
diff --git a/python/sdist/amici/petab/sbml_import.py b/python/sdist/amici/petab/sbml_import.py
index 6c9141f908..8443639a8a 100644
--- a/python/sdist/amici/petab/sbml_import.py
+++ b/python/sdist/amici/petab/sbml_import.py
@@ -135,6 +135,7 @@ def import_model_sbml(
             else SbmlModel.from_file(sbml_model),
             condition_df=petab.get_condition_df(condition_table),
             observable_df=petab.get_observable_df(observable_table),
+            measurement_df=petab.get_measurement_df(measurement_table),
         )
 
     if petab_problem.observable_df is None: