diff --git a/packages/frontend/core/src/modules/feature-flag/constant.ts b/packages/frontend/core/src/modules/feature-flag/constant.ts
index 7b5543462..04ea7e1b8 100644
--- a/packages/frontend/core/src/modules/feature-flag/constant.ts
+++ b/packages/frontend/core/src/modules/feature-flag/constant.ts
@@ -209,8 +209,8 @@ export const AFFINE_FLAGS = {
     category: 'affine',
     displayName: 'Enable AI Button',
     description: 'Enable AI Button on mobile',
-    configurable: BUILD_CONFIG.isMobileEdition && isCanaryBuild,
-    defaultState: false,
+    configurable: BUILD_CONFIG.isMobileEdition && BUILD_CONFIG.isIOS,
+    defaultState: BUILD_CONFIG.isMobileEdition && BUILD_CONFIG.isIOS,
   },
   enable_turbo_renderer: {
     category: 'blocksuite',