Revise the getRoles method