From 43dc0481554f708a9520bd5c0e9d1481d913bebc Mon Sep 17 00:00:00 2001
From: Danny White <3104761+dnywh@users.noreply.github.com>
Date: Wed, 24 Sep 2025 08:38:17 +1000
Subject: [PATCH] fix: invalidate upgrade eligibility query (#38907)

* invalidate query and add useEffect

* revert useEffect
---
 .../database-extension-enable-mutation.ts                   | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/apps/studio/data/database-extensions/database-extension-enable-mutation.ts b/apps/studio/data/database-extensions/database-extension-enable-mutation.ts
index 68d74ad972..f7b7719829 100644
--- a/apps/studio/data/database-extensions/database-extension-enable-mutation.ts
+++ b/apps/studio/data/database-extensions/database-extension-enable-mutation.ts
@@ -4,6 +4,7 @@ import { useMutation, UseMutationOptions, useQueryClient } from '@tanstack/react
 import { toast } from 'sonner'
 
 import { executeSql } from 'data/sql/execute-sql-query'
+import { configKeys } from 'data/config/keys'
 import type { ResponseError } from 'types'
 import { databaseExtensionsKeys } from './keys'
 
@@ -57,7 +58,10 @@ export const useDatabaseExtensionEnableMutation = ({
     {
       async onSuccess(data, variables, context) {
         const { projectRef } = variables
-        await queryClient.invalidateQueries(databaseExtensionsKeys.list(projectRef))
+        await Promise.all([
+          queryClient.invalidateQueries(databaseExtensionsKeys.list(projectRef)),
+          queryClient.invalidateQueries(configKeys.upgradeEligibility(projectRef)),
+        ])
         await onSuccess?.(data, variables, context)
       },
       async onError(data, variables, context) {