diff --git a/cognee/api/v1/permissions/routers/get_permissions_router.py b/cognee/api/v1/permissions/routers/get_permissions_router.py
index db2c72705..20d35e748 100644
--- a/cognee/api/v1/permissions/routers/get_permissions_router.py
+++ b/cognee/api/v1/permissions/routers/get_permissions_router.py
@@ -259,7 +259,7 @@ def get_permissions_router() -> APIRouter:
 
         from cognee.modules.users.tenants.methods import select_tenant as select_tenant_method
 
-        await select_tenant_method(user=user, tenant_id=payload.tenant_id)
+        await select_tenant_method(user_id=user.id, tenant_id=payload.tenant_id)
 
         return JSONResponse(
             status_code=200,
diff --git a/cognee/modules/users/tenants/methods/select_tenant.py b/cognee/modules/users/tenants/methods/select_tenant.py
index cb291d5f2..83c11dc91 100644
--- a/cognee/modules/users/tenants/methods/select_tenant.py
+++ b/cognee/modules/users/tenants/methods/select_tenant.py
@@ -5,19 +5,20 @@ import sqlalchemy.exc
 from sqlalchemy import select
 
 from cognee.infrastructure.databases.relational import get_relational_engine
+from cognee.modules.users.methods.get_user import get_user
 from cognee.modules.users.models.UserTenant import UserTenant
 from cognee.modules.users.models.User import User
 from cognee.modules.users.permissions.methods import get_tenant
 from cognee.modules.users.exceptions import UserNotFoundError, TenantNotFoundError
 
 
-async def select_tenant(user: User, tenant_id: Union[UUID, None]) -> User:
+async def select_tenant(user_id: UUID, tenant_id: Union[UUID, None]) -> User:
     """
         Set the users active tenant to provided tenant.
 
         If None tenant_id is provided set current Tenant to the default single user-tenant
     Args:
-        user: User object.
+        user_id: UUID of the user.
         tenant_id: Id of the tenant.
 
     Returns:
@@ -26,6 +27,7 @@ async def select_tenant(user: User, tenant_id: Union[UUID, None]) -> User:
     """
     db_engine = get_relational_engine()
     async with db_engine.get_async_session() as session:
+        user = await get_user(user_id)
         if tenant_id is None:
             # If no tenant_id is provided set current Tenant to the single user-tenant
             user.tenant_id = None
diff --git a/examples/python/permissions_example.py b/examples/python/permissions_example.py
index 4bbd30bea..c0b104023 100644
--- a/examples/python/permissions_example.py
+++ b/examples/python/permissions_example.py
@@ -145,7 +145,7 @@ async def main():
     tenant_id = await create_tenant("CogneeLab", user_2.id)
 
     print("User 2 is selecting CogneeLab tenant/organization as active tenant")
-    await select_tenant(user=user_2, tenant_id=tenant_id)
+    await select_tenant(user_id=user_2.id, tenant_id=tenant_id)
 
     print("\nUser 2 is creating Researcher role")
     role_id = await create_role(role_name="Researcher", owner_id=user_2.id)
@@ -163,7 +163,7 @@ async def main():
     await add_user_to_role(user_id=user_3.id, role_id=role_id, owner_id=user_2.id)
 
     print("\nOperation as user_3 to select CogneeLab tenant/organization as active tenant")
-    await select_tenant(user=user_3, tenant_id=tenant_id)
+    await select_tenant(user_id=user_3.id, tenant_id=tenant_id)
 
     print(
         "\nOperation started as user_2, with CogneeLab as its active tenant, to give read permission to Researcher role for the dataset QUANTUM owned by user_2"
@@ -187,6 +187,10 @@ async def main():
     )
     # We can re-create the QUANTUM dataset in the CogneeLab tenant. The old QUANTUM dataset is still owned by user_2 personally
     # and can still be accessed by selecting the personal tenant for user 2.
+    from cognee.modules.users.methods import get_user
+
+    # Note: We need to update user_2 from the database to refresh its tenant context changes
+    user_2 = await get_user(user_2.id)
     await cognee.add([text], dataset_name="QUANTUM_COGNEE_LAB", user=user_2)
     quantum_cognee_lab_cognify_result = await cognee.cognify(["QUANTUM_COGNEE_LAB"], user=user_2)