diff --git a/packages/frontend/core/src/desktop/dialogs/setting/account-setting/ai-usage-panel.tsx b/packages/frontend/core/src/desktop/dialogs/setting/account-setting/ai-usage-panel.tsx
index af5b07a6f5..357adf3595 100644
--- a/packages/frontend/core/src/desktop/dialogs/setting/account-setting/ai-usage-panel.tsx
+++ b/packages/frontend/core/src/desktop/dialogs/setting/account-setting/ai-usage-panel.tsx
@@ -31,6 +31,7 @@ export const AIUsagePanel = ({
   useEffect(() => {
     // revalidate latest subscription status
     subscriptionService.subscription.revalidate();
+    subscriptionService.prices.revalidate();
   }, [subscriptionService]);
   const copilotQuotaService = useService(UserCopilotQuotaService);
   useEffect(() => {
diff --git a/packages/frontend/core/src/desktop/dialogs/setting/general-setting/plans/ai/actions/subscribe.tsx b/packages/frontend/core/src/desktop/dialogs/setting/general-setting/plans/ai/actions/subscribe.tsx
index 59da7eca76..0d501a30fa 100644
--- a/packages/frontend/core/src/desktop/dialogs/setting/general-setting/plans/ai/actions/subscribe.tsx
+++ b/packages/frontend/core/src/desktop/dialogs/setting/general-setting/plans/ai/actions/subscribe.tsx
@@ -74,7 +74,12 @@ export const AISubscribe = ({
       onBeforeCheckout={onBeforeCheckout}
       checkoutOptions={checkoutOptions}
       renderer={props => (
-        <Button variant="primary" {...props} {...btnProps}>
+        <Button
+          variant="primary"
+          {...props}
+          {...btnProps}
+          data-testid="ai-subscribe-button"
+        >
           {btnProps.children ?? `${priceReadable} / ${priceFrequency}`}
           {displayedFrequency === 'monthly' ? (
             <span