diff --git a/notebooks/abliterate.ipynb b/notebooks/abliterate.ipynb
index e5085a5..a89c6e5 100644
--- a/notebooks/abliterate.ipynb
+++ b/notebooks/abliterate.ipynb
@@ -136,9 +136,9 @@
     "def on_stage(stage):\n",
     "    icons = {\"summon\": \"\\u26a1\", \"probe\": \"\\u2692\", \"distill\": \"\\u269b\",\n",
     "             \"excise\": \"\\u2702\", \"verify\": \"\\u2713\", \"rebirth\": \"\\u2606\"}\n",
-    "    icon = icons.get(stage.key, \"\")\n",
+    "    icon = icons.get(stage.stage, \"\")\n",
     "    print(f\"\\n{'='*60}\")\n",
-    "    print(f\"{icon}  STAGE: {stage.key.upper()} — {stage.description}\")\n",
+    "    print(f\"{icon}  STAGE: {stage.stage.upper()} — {stage.message}\")\n",
     "    print(f\"{'='*60}\")\n",
     "\n",
     "def on_log(msg):\n",
@@ -152,7 +152,7 @@
     "\n",
     "print(f\"\\n{'='*60}\")\n",
     "print(f\"ABLITERATION COMPLETE\")\n",
-    "print(f\"Output: {result.get('output_dir', OUTPUT_DIR)}\")\n",
+    "print(f\"Output: {result}\")\n",
     "print(f\"{'='*60}\")"
    ]
   },
@@ -295,4 +295,4 @@
    ]
   }
  ]
-}
\ No newline at end of file
+}