diff --git a/odoo-stubs/tools/template_inheritance.pyi b/odoo-stubs/tools/template_inheritance.pyi
index 34b7a6e..7cbc797 100644
--- a/odoo-stubs/tools/template_inheritance.pyi
+++ b/odoo-stubs/tools/template_inheritance.pyi
@@ -1,9 +1,10 @@
-from typing import Any
+from typing import Any, Callable
 
-_logger: Any
+from lxml.etree import _Element
 
-def add_text_before(node, text) -> None: ...
-def add_text_inside(node, text) -> None: ...
-def remove_element(node) -> None: ...
-def locate_node(arch, spec): ...
-def apply_inheritance_specs(source, specs_tree, inherit_branding: bool = ..., pre_locate=...): ...
+def add_text_before(node: _Element, text: str | None) -> None: ...
+def add_text_inside(node: _Element, text: str | None) -> None: ...
+def remove_element(node: _Element) -> None: ...
+def locate_node(arch: _Element, spec: _Element) -> _Element: ...
+def apply_inheritance_specs(source: _Element, specs_tree: _Element, inherit_branding: bool = ...,
+                            pre_locate: Callable[[_Element], Any] = ...) -> _Element: ...