diff --git a/python/sdist/amici/petab_objective.py b/python/sdist/amici/petab_objective.py
index af4e7172c1..0e54fb022d 100644
--- a/python/sdist/amici/petab_objective.py
+++ b/python/sdist/amici/petab_objective.py
@@ -7,6 +7,7 @@
     DeprecationWarning,
 )
 
+from .petab.simulations import create_edatas  # noqa: F401
 from .petab.simulations import (  # noqa: F401
     aggregate_sllh,
     rdatas_to_measurement_df,
diff --git a/tests/benchmark-models/test_petab_benchmark.py b/tests/benchmark-models/test_petab_benchmark.py
index 9d122b8570..af9dfc9a88 100755
--- a/tests/benchmark-models/test_petab_benchmark.py
+++ b/tests/benchmark-models/test_petab_benchmark.py
@@ -53,7 +53,7 @@
 
 # until fiddy is updated
 @pytest.mark.filterwarnings(
-    "ignore:Importing amici.petab_objective is deprecated."
+    "ignore:Importing amici.petab_objective is deprecated.*:DeprecationWarning"
 )
 @pytest.mark.filterwarnings("ignore:divide by zero encountered in log10")
 @pytest.mark.parametrize("scale", (True, False))