diff --git a/s07_skill_loading/code.py b/s07_skill_loading/code.py
index fd3a5d9..40d8680 100644
--- a/s07_skill_loading/code.py
+++ b/s07_skill_loading/code.py
@@ -344,13 +344,10 @@ def agent_loop(messages: list):
global rounds_since_todo
while True:
if rounds_since_todo >= 3 and messages:
- last = messages[-1]
- if last["role"] == "user" and isinstance(last.get("content"), list):
- last["content"].insert(0, {
- "type": "text",
- "text": "Update your todos.",
- })
-
+ messages.append({"role": "user",
+ "content": "Update your todos."})
+ rounds_since_todo = 0
+
response = client.messages.create(
model=MODEL, system=SYSTEM, messages=messages,
tools=TOOLS, max_tokens=8000,