diff --git a/packages/extension/src/ui/features/accounts/upgrade.service.ts b/packages/extension/src/ui/features/accounts/upgrade.service.ts
index 4deae5083..a4f3d2a05 100644
--- a/packages/extension/src/ui/features/accounts/upgrade.service.ts
+++ b/packages/extension/src/ui/features/accounts/upgrade.service.ts
@@ -224,7 +224,6 @@ export const useCheckUpgradeAvailable = (account?: Account) => {
     () => account && checkIfUpgradeAvailable(account, accountClassHash),
     { suspense: false },
   )
-  return { needsUpgrade: false, needsUpgradeError, needsUpgradeValidating }
 
-  // return { needsUpgrade, needsUpgradeError, needsUpgradeValidating }
+  return { needsUpgrade, needsUpgradeError, needsUpgradeValidating }
 }