diff --git a/GramAddict/core/perception/action_memory.py b/GramAddict/core/perception/action_memory.py
index b45f51d..5185f82 100644
--- a/GramAddict/core/perception/action_memory.py
+++ b/GramAddict/core/perception/action_memory.py
@@ -234,7 +234,7 @@ class ActionMemory:
                     f"If the intent was to open a profile, are you on a profile page? "
                     f"If the intent was to go back, are you on the previous screen? "
                 )
-            prompt += "Answer ONLY with the word YES or NO."
+            prompt += 'Answer ONLY with a valid JSON object exactly matching this schema: {"success": true} or {"success": false}. DO NOT add any other keys.'
 
             try:
                 screenshot = device.get_screenshot_b64()