Post login roles with enterprise Google workspace authentication

Ready to post? :mag: First, try searching for your answer.
Hello,
I have set up enterprise Google workplace authentication and under action I have set up post-login actions to add claims. When I inspect the JWT I do not see any roles. My post-login action works without Google Workspace authentication. Do you have any idea how to fix this issue?

Thanks