diff --git a/examples/rewoo/rewoo.ipynb b/examples/rewoo/rewoo.ipynb
index 218713d2e..d4f2c7aff 100644
--- a/examples/rewoo/rewoo.ipynb
+++ b/examples/rewoo/rewoo.ipynb
@@ -342,6 +342,7 @@
     "        _results = state[\"results\"] or {}\n",
     "        for k, v in _results.items():\n",
     "            tool_input = tool_input.replace(k, v)\n",
+    "            step_name = step_name.replace(k, v)\n",
     "        plan += f\"Plan: {_plan}\\n{step_name} = {tool}[{tool_input}]\"\n",
     "    prompt = solve_prompt.format(plan=plan, task=state[\"task\"])\n",
     "    result = model.invoke(prompt)\n",