diff --git a/studio/data/profile/profile-create-mutation.ts b/studio/data/profile/profile-create-mutation.ts index a6a3f358a4c..37ff0d23424 100644 --- a/studio/data/profile/profile-create-mutation.ts +++ b/studio/data/profile/profile-create-mutation.ts @@ -1,6 +1,8 @@ import { useMutation, UseMutationOptions, useQueryClient } from '@tanstack/react-query' import { toast } from 'react-hot-toast' +import { organizationKeys } from 'data/organizations/keys' +import { permissionKeys } from 'data/permissions/keys' import { post } from 'lib/common/fetch' import { API_URL } from 'lib/constants' import { ResponseError } from 'types' @@ -29,7 +31,11 @@ export const useProfileCreateMutation = ({ return useMutation(() => createProfile(), { async onSuccess(data, variables, context) { - await queryClient.invalidateQueries(profileKeys.profile()) + await Promise.all([ + queryClient.invalidateQueries(profileKeys.profile()), + queryClient.invalidateQueries(organizationKeys.list()), + queryClient.invalidateQueries(permissionKeys.list()), + ]) await onSuccess?.(data, variables, context) }, async onError(data, variables, context) { diff --git a/studio/lib/profile.tsx b/studio/lib/profile.tsx index d8feec8dcbc..5810bce5301 100644 --- a/studio/lib/profile.tsx +++ b/studio/lib/profile.tsx @@ -3,7 +3,6 @@ import { useRouter } from 'next/router' import { PropsWithChildren, createContext, useContext, useMemo } from 'react' import { useTelemetryProps } from 'common' -import { invalidateOrganizationsQuery } from 'data/organizations/organizations-query' import { usePermissionsQuery } from 'data/permissions/permissions-query' import { useProfileCreateMutation } from 'data/profile/profile-create-mutation' import { useProfileQuery } from 'data/profile/profile-query' @@ -41,8 +40,6 @@ export const ProfileProvider = ({ children }: PropsWithChildren<{}>) => { telemetryProps, router ) - - await invalidateOrganizationsQuery(queryClient) }, onError() { ui.setNotification({