Trigger does not execute function on Update existing table row

BEGIN
  -- Update the custom claim for onboarding
  PERFORM public.set_claim(
    NEW.id, 
    'onboarding'::text, 
    to_jsonb(NEW.onboarding)
  );

  RETURN NULL;
END;

CREATE OR REPLACE TRIGGER trigger_users_onboarding_updated
AFTER UPDATE OF onboarding ON public.users
FOR EACH ROW
EXECUTE FUNCTION public.update_user_onboarding_claim();