diff --git a/linopy/constants.py b/linopy/constants.py
index 0e971827..11194407 100644
--- a/linopy/constants.py
+++ b/linopy/constants.py
@@ -156,7 +156,7 @@ def process(cls, status: str) -> Self:
     @classmethod
     def from_termination_condition(
         cls, termination_condition: "TerminationCondition"
-    ) -> "SolverStatus":
+    ) -> Self:
         for status in STATUS_TO_TERMINATION_CONDITION_MAP:
             if termination_condition in STATUS_TO_TERMINATION_CONDITION_MAP[status]:
                 return status