diff --git a/Video_Killed_The_Radio_Star_Defusion.ipynb b/Video_Killed_The_Radio_Star_Defusion.ipynb
index 5f25a2d..cc89f10 100644
--- a/Video_Killed_The_Radio_Star_Defusion.ipynb
+++ b/Video_Killed_The_Radio_Star_Defusion.ipynb
@@ -715,6 +715,11 @@
         "\n",
         "    # I give up.\n",
         "    def get_image_for_prompt(*args, **kargs):\n",
+        "        # ugly hotfix. todo: unify function signatures...\n",
+        "        if 'init_image' in kargs:\n",
+        "            img = kargs.pop('init_image')\n",
+        "            if img:\n",
+        "                kargs['image'] = img\n",
         "        result = hf_helper.get_image_for_prompt(*args, **kargs)\n",
         "        return result.images\n",
         "\n",