diff --git a/python/sdist/amici/petab/cli/import_petab.py b/python/sdist/amici/petab/cli/import_petab.py
index b124b4d98b..d61cda5362 100644
--- a/python/sdist/amici/petab/cli/import_petab.py
+++ b/python/sdist/amici/petab/cli/import_petab.py
@@ -1,6 +1,6 @@
 import argparse
 
-import petab
+import petab.v1 as petab
 
 from ..petab_import import import_model_sbml
 from petab.v1.models.sbml_model import SbmlModel
diff --git a/tests/performance/test.py b/tests/performance/test.py
index f117371910..677e6e647d 100755
--- a/tests/performance/test.py
+++ b/tests/performance/test.py
@@ -8,8 +8,8 @@
 from pathlib import Path
 
 import amici
-import petab
-from amici.petab_import import import_model
+import petab.v1 as petab
+from amici.petab.petab_import import import_model
 
 
 def parse_args():