diff --git a/src/repository/users.py b/src/repository/users.py
index 3ada358..def44c9 100644
--- a/src/repository/users.py
+++ b/src/repository/users.py
@@ -1,5 +1,6 @@
 from sqlalchemy import delete, select, update
 
+from schemas.users import User
 from src.models import UsersORM
 from src.repository.base import BaseRepo
 
@@ -7,7 +8,7 @@ from src.repository.base import BaseRepo
 class UsersRepo(BaseRepo):
     model = UsersORM
 
-    async def get_all_users(self):
+    async def get_all_users(self) -> list[UsersORM]:
         query = select(self.model)
         result = await self.session.execute(query)
         models = result.scalars().all()
@@ -16,7 +17,7 @@ class UsersRepo(BaseRepo):
     async def delete_one(self, id: int) -> None:
         await self.session.execute(delete(self.model).where(self.model.id == id))
 
-    async def update_one(self, id: int, data: dict) -> None:
+    async def update_one(self, id: int, data: dict) -> UsersORM:
         stmt = update(self.model).where(self.model.id == id).values(data.model_dump(exclude_unset=True)).returning(self.model)
         result = await self.session.execute(stmt)
         model = result.scalar_one()