diff --git a/backend/src/bot/modules/ai/prompts.py b/backend/src/bot/modules/ai/prompts.py
index a5b6743..b1e9bca 100644
--- a/backend/src/bot/modules/ai/prompts.py
+++ b/backend/src/bot/modules/ai/prompts.py
@@ -16,7 +16,9 @@ EXAM_FOLLOW_UP = """You see a problem set image. List available problem numbers.
 Output only the numbers that exist in the image, like: 1, 2, 3, 4, 5
 If problems have letters (a, b, c), list them as: 1a, 1b, 2a, etc.
 Keep it minimal - just the identifiers.
-Then, if applicable, output some possible followups of conversation"""
+Then, if applicable, output some possible followups of conversation
+If there is a problem set, you MUST output ALL of problem numbers/letters,
+before another followups"""
 
 DEFAULT_FOLLOW_UP = (
     "Based on the conversation, suggest 3 short follow-up questions "