diff --git a/notebooks/comfyui_colab.ipynb b/notebooks/comfyui_colab.ipynb
index 4762c03c..cdf182b8 100644
--- a/notebooks/comfyui_colab.ipynb
+++ b/notebooks/comfyui_colab.ipynb
@@ -106,8 +106,10 @@
         "        break\n",
         "      sock.close()\n",
         "  from google.colab import output\n",
-        "  x = output.serve_kernel_port_as_window(port)\n",
-        "  x = output.serve_kernel_port_as_iframe(port, height=1024)\n",
+        "  output.serve_kernel_port_as_iframe(port, height=1024)\n",
+        "  print(\"to open it in a window you can open this link here:\")\n",
+        "  output.serve_kernel_port_as_window(port)\n",
+        "\n",
         "threading.Thread(target=iframe_thread, daemon=True, args=(8188,)).start()\n",
         "\n",
         "!python main.py"