diff --git a/superflexpy/implementation/models/thur_M2.py b/superflexpy/implementation/models/thur_M2.py
index 9735159..9f5a901 100644
--- a/superflexpy/implementation/models/thur_M2.py
+++ b/superflexpy/implementation/models/thur_M2.py
@@ -36,6 +36,7 @@
 from superflexpy.framework.network import Network
 from superflexpy.framework.node import Node
 from superflexpy.framework.unit import Unit
+from superflexpy.implementation.elements.hbv import PowerReservoir, UnsaturatedReservoir
 from superflexpy.implementation.elements.structure_elements import (
     Junction,
     Splitter,
@@ -43,9 +44,7 @@
 )
 from superflexpy.implementation.elements.thur_model_hess import (
     HalfTriangularLag,
-    PowerReservoir,
     SnowReservoir,
-    UnsaturatedReservoir,
 )
 from superflexpy.implementation.numerical_approximators.implicit_euler import (
     ImplicitEulerPython,