diff --git a/odoo-stubs/modules/module.pyi b/odoo-stubs/modules/module.pyi
index 48dc48a..f452f09 100644
--- a/odoo-stubs/modules/module.pyi
+++ b/odoo-stubs/modules/module.pyi
@@ -38,6 +38,7 @@ def get_modules_with_version() -> dict[str, Any]: ...
 def adapt_version(version: str) -> str: ...
 def get_test_modules(module: str) -> list: ...
 def _get_tests_modules(path: str, module: str) -> list: ...
+def _get_upgrade_test_modules(module) -> Generator[ModuleType, None, None]: ...
 
 class OdooTestResult(unittest.result.TestResult):
     def log(self, level, msg, *args, test: Any | None = ..., exc_info: Any | None = ..., extra: Any | None = ..., stack_info: bool = ..., caller_infos: Any | None = ...) -> None: ...