diff --git a/frontend/appflowy_flutter/lib/plugins/ai_chat/application/chat_input_bloc.dart b/frontend/appflowy_flutter/lib/plugins/ai_chat/application/chat_input_bloc.dart
index 8a6bdf55be..1d7bf4f589 100644
--- a/frontend/appflowy_flutter/lib/plugins/ai_chat/application/chat_input_bloc.dart
+++ b/frontend/appflowy_flutter/lib/plugins/ai_chat/application/chat_input_bloc.dart
@@ -52,7 +52,7 @@ class ChatInputBloc extends Bloc<ChatInputEvent, ChatInputState> {
         );
       },
       updateState: (aiState) {
-        if (aiState.enabled) {
+        if (aiState.pluginState.state == RunningStatePB.Running) {
           emit(const ChatInputState(aiType: _LocalAI()));
         } else {
           emit(const ChatInputState(aiType: _AppFlowyAI()));