diff --git a/deepcave/plugins/hyperparameter/ablation_paths.py b/deepcave/plugins/hyperparameter/ablation_paths.py
index 167aa0b0..d5be095a 100644
--- a/deepcave/plugins/hyperparameter/ablation_paths.py
+++ b/deepcave/plugins/hyperparameter/ablation_paths.py
@@ -212,7 +212,7 @@ def load_inputs(self) -> Dict[str, Dict[str, Any]]:
             "show_confidence": {"options": get_select_options(binary=True), "value": "false"},
         }
 
-    def load_dependency_inputs(self, run, _: Any, inputs: Dict[str, Any]) -> Dict[
+    def load_dependency_inputs(self, run: AbstractRun, _: Any, inputs: Dict[str, Any]) -> Dict[
         str, Any]:  # type: ignore # noqa: E501
         """
         Works like 'load_inputs' but called after inputs have changed.
@@ -224,7 +224,7 @@ def load_dependency_inputs(self, run, _: Any, inputs: Dict[str, Any]) -> Dict[
 
         Parameters
         ----------
-        run
+        run: AbstractRun
             The selected run.
         inputs : Dict[str, Any]
             Current content of the inputs.