diff --git a/src/moscot/base/problems/problem.py b/src/moscot/base/problems/problem.py
index 475309561..eb0a340de 100644
--- a/src/moscot/base/problems/problem.py
+++ b/src/moscot/base/problems/problem.py
@@ -14,6 +14,7 @@
 )
 
 import cloudpickle
+from docstring_inheritance import NumpyDocstringInheritanceMeta
 
 import numpy as np
 import pandas as pd
@@ -39,7 +40,7 @@
 )
 from moscot.base.solver import OTSolver
 from moscot.utils.tagged_array import Tag, TaggedArray
-from docstring_inheritance import NumpyDocstringInheritanceMeta
+
 __all__ = ["BaseProblem", "OTProblem"]