diff --git a/odoo-stubs/__init__.pyi b/odoo-stubs/__init__.pyi
index c8698cf..5976a53 100644
--- a/odoo-stubs/__init__.pyi
+++ b/odoo-stubs/__init__.pyi
@@ -1,16 +1,33 @@
-from . import addons as addons, api as api, cli as cli, conf as conf, fields as fields, http as http, loglevels as loglevels, models as models, netsvc as netsvc, osv as osv, release as release, service as service, sql_db as sql_db, tools as tools, upgrade as upgrade
-from .tools.translate import _ as _, _lt as _lt
-from typing import Any, Optional
+from psycopg2 import connection
+
+from . import (
+    addons as addons,
+    api as api,
+    cli as cli,
+    conf as conf,
+    fields as fields,
+    http as http,
+    loglevels as loglevels,
+    models as models,
+    netsvc as netsvc,
+    osv as osv,
+    release as release,
+    service as service,
+    sql_db as sql_db,
+    tools as tools,
+    upgrade as upgrade
+)
+from .api import Registry
+from .tools.translate import _ as _, _lt as _lt
 
-__path__: Any
 evented: bool
 
-def gevent_wait_callback(conn: Any, timeout: Optional[Any] = ...) -> None: ...
+def gevent_wait_callback(conn: connection, timeout: float | None = ...) -> None: ...
 
 multi_process: bool
 
-def _decompress(data: Any): ...
+def _decompress(data: bytes) -> bytes: ...
 
 SUPERUSER_ID: int
 
-def registry(database_name: Optional[Any] = ...): ...
+def registry(database_name: str | None = ...) -> Registry: ...