diff --git a/notify-api/src/notify_api/__init__.py b/notify-api/src/notify_api/__init__.py
index ba7808b5d..5fd1feca2 100644
--- a/notify-api/src/notify_api/__init__.py
+++ b/notify-api/src/notify_api/__init__.py
@@ -44,7 +44,7 @@ def setup_jwt_manager(app):
     """Use flask app to configure the JWTManager to work for a particular Realm."""
 
     def get_roles(a_dict):
-        return a_dict['client_roles']  # pragma: no cover
+        return a_dict['realm_access']['roles']  # pragma: no cover
 
     app.config['JWT_ROLE_CALLBACK'] = get_roles