diff --git a/packages/create/src/frameworks/react/add-ons/ai/assets/src/components/demo-AIAssistant.tsx b/packages/create/src/frameworks/react/add-ons/ai/assets/src/components/demo-AIAssistant.tsx
index 851ba716..b896cbeb 100644
--- a/packages/create/src/frameworks/react/add-ons/ai/assets/src/components/demo-AIAssistant.tsx
+++ b/packages/create/src/frameworks/react/add-ons/ai/assets/src/components/demo-AIAssistant.tsx
@@ -79,7 +79,7 @@ function Messages({ messages }: { messages: ChatMessages }) {
 }
 
 export default function AIAssistant() {
-  const isOpen = useStore(showAIAssistant)
+  const isOpen = useStore(showAIAssistant, (state) => state)
   const { messages, sendMessage } = useGuitarRecommendationChat()
   const [input, setInput] = useState('')