From 7d05bb6dac6a57cdeabb31800e248394c12f6cee Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Mon, 13 Nov 2023 06:52:00 +0300
Subject: Update check for admins

---
 gn_auth/auth/authorisation/users/admin/ui.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'gn_auth/auth/authorisation/users/admin')

diff --git a/gn_auth/auth/authorisation/users/admin/ui.py b/gn_auth/auth/authorisation/users/admin/ui.py
index 68e1ba5..64e79a0 100644
--- a/gn_auth/auth/authorisation/users/admin/ui.py
+++ b/gn_auth/auth/authorisation/users/admin/ui.py
@@ -3,12 +3,12 @@ from functools import wraps
 from flask import flash, url_for, redirect
 
 from gn_auth.session import logged_in, session_user, clear_session_info
+from gn_auth.auth.authorisation.resources.system.models import (
+    user_roles_on_system)
 
 from ....authentication.users import User
 from ....db.sqlite3 import with_db_connection
 
-from ...roles.models import user_roles
-
 def is_admin(func):
     """Verify user is a system admin."""
     @wraps(func)
@@ -16,7 +16,7 @@ def is_admin(func):
     def __admin__(*args, **kwargs):
         admin_roles = [
             role for role in with_db_connection(
-                lambda conn: user_roles(
+                lambda conn: user_roles_on_system(
                     conn, User(**session_user())))
             if role.role_name == "system-administrator"]
         if len(admin_roles) > 0:
-- 
cgit v1.2.3