Execute Authorization Policy API isnt working as expected

We are using Authorisation extension for managing groups/roles.permissions.
We have mappings from Azure IDP. All synchronisation and policies were working fine.
We implemented Organization feature for B2B. Now the Authoriztion extension seem to be broken.
We manually tried the API “Execute Authorization Policy” (Authorization Extension API Explorer) manually to enforce the mappings, the response seems to be correct and contains all the groups,roles and permissions. But the same is not applied within authz extension. Groups and roles are not assigned with the api.
As said earlier, the same setup was working fine before and after enabling organization, the same is not working.
Any inputs are appreciated.