diff --git a/odoo-stubs/models.pyi b/odoo-stubs/models.pyi
index d212ac2..af2a739 100644
--- a/odoo-stubs/models.pyi
+++ b/odoo-stubs/models.pyi
@@ -17,7 +17,7 @@ from typing import (
 
 import psycopg2
 from odoo.addons.base.models.res_company import Company
-from odoo.addons.base.models.res_partner import Partner
+from odoo.addons.base.models.res_users import Users
 
 from . import api, fields
 from .api import Environment
@@ -447,7 +447,7 @@ class BaseModel(metaclass=MetaModel):
     def ensure_one(self: _ModelT) -> _ModelT: ...
     def with_env(self: _ModelT, env: Environment) -> _ModelT: ...
     def sudo(self: _ModelT, flag: bool = ...) -> _ModelT: ...
-    def with_user(self: _ModelT, user: Partner | int) -> _ModelT: ...
+    def with_user(self: _ModelT, user: Users | int) -> _ModelT: ...
     def with_company(self: _ModelT, company: Company | int) -> _ModelT: ...
     def with_context(self: _ModelT, *args, **kwargs) -> _ModelT: ...
     def with_prefetch(