diff --git a/python/sdist/amici/sim/sundials/__init__.py b/python/sdist/amici/sim/sundials/__init__.py
index 9f64e9be66..1d8e2209c4 100644
--- a/python/sdist/amici/sim/sundials/__init__.py
+++ b/python/sdist/amici/sim/sundials/__init__.py
@@ -53,6 +53,7 @@
     #  from .swig_wrappers
     hdf5_enabled = "read_solver_settings_from_hdf5" in dir()
     # These modules require the swig interface and other dependencies
+    from ._debugging import get_model_for_preeq
     from ._numpy import ExpDataView as ExpDataView
     from ._numpy import ReturnDataView as ReturnDataView
     from ._numpy import evaluate as evaluate
@@ -83,6 +84,7 @@ def get_model(self) -> amici_swig_py.Model:
         "ModelModule",
         *_swig_wrappers.__all__,
         "AmiciModel",
+        "get_model_for_preeq",
     ]
 
     # expose the swig module itself
diff --git a/python/sdist/amici/debugging/__init__.py b/python/sdist/amici/sim/sundials/_debugging.py
similarity index 100%
rename from python/sdist/amici/debugging/__init__.py
rename to python/sdist/amici/sim/sundials/_debugging.py
diff --git a/python/tests/test_preequilibration.py b/python/tests/test_preequilibration.py
index af73f5749b..f33156a811 100644
--- a/python/tests/test_preequilibration.py
+++ b/python/tests/test_preequilibration.py
@@ -6,7 +6,6 @@
 import amici
 import numpy as np
 import pytest
-from amici.debugging import get_model_for_preeq
 from amici.importers.antimony import antimony2amici
 from amici.sim.sundials import (
     AMICI_ERROR,
@@ -19,6 +18,7 @@
     SteadyStateComputationMode,
     SteadyStateSensitivityMode,
     SteadyStateStatus,
+    get_model_for_preeq,
     parameter_scaling_from_int_vector,
     run_simulation,
 )