diff --git a/odoo-stubs/modules/loading.pyi b/odoo-stubs/modules/loading.pyi
index 9e0e2c4..3f0a404 100644
--- a/odoo-stubs/modules/loading.pyi
+++ b/odoo-stubs/modules/loading.pyi
@@ -2,7 +2,7 @@ from typing import Any, Iterable
 
 from .graph import Graph, Node
 from .registry import Registry
-from ..sql_db import Cursor
+from ..sql_db import Cursor, Connection
 from ..tests.runner import OdooTestResult
 
 def load_data(cr: Cursor, idref: dict, mode: str, kind: str, package: Node) -> bool: ...
@@ -13,5 +13,5 @@ def load_module_graph(cr: Cursor, graph: Graph, status: Any | None = ..., perfor
 def _check_module_names(cr: Cursor, module_names: Iterable[str]) -> None: ...
 def load_marked_modules(cr: Cursor, graph: Graph, states: list[str], force: list[str], progressdict, report: OdooTestResult,
                         loaded_modules: list[str], perform_checks: Any, models_to_check: set[str] | None = ...) -> list[str]: ...
-def load_modules(db: Registry, force_demo: bool = ..., status: Any | None = ..., update_module: Any = ...) -> Registry | None: ...
+def load_modules(db: Connection, force_demo: bool = ..., status: Any | None = ..., update_module: Any = ...) -> Registry | None: ...
 def reset_modules_state(db_name: str) -> None: ...