diff --git a/notebooks/9-FINN-EndToEndFlow.ipynb b/notebooks/9-FINN-EndToEndFlow.ipynb
index 73049e0967c10c107bdd1f374013cdd12df9e567..18c1ead92e702aed8bca1bd01f67b81694861709 100644
--- a/notebooks/9-FINN-EndToEndFlow.ipynb
+++ b/notebooks/9-FINN-EndToEndFlow.ipynb
@@ -13,11 +13,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 34,
    "metadata": {},
    "outputs": [],
    "source": [
     "import inspect\n",
+    "import netron\n",
     "\n",
     "def showSrc(what):\n",
     "    print(\"\".join(inspect.getsourcelines(what)[0]))"
@@ -128,7 +129,6 @@
     }
    ],
    "source": [
-    "import netron\n",
     "netron.start(\"lfc_w1_a1.onnx\", port=8081, host=\"0.0.0.0\")"
    ]
   },
@@ -790,7 +790,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 32,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -808,15 +808,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 58,
+   "execution_count": 35,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\n",
-      "Stopping http://0.0.0.0:8081\n",
       "Serving 'lfc_w1_a1.onnx' at http://0.0.0.0:8081\n"
      ]
     }
@@ -828,8 +826,10 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 59,
-   "metadata": {},
+   "execution_count": 36,
+   "metadata": {
+    "scrolled": true
+   },
    "outputs": [
     {
      "data": {
@@ -862,7 +862,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 21,
+   "execution_count": 37,
    "metadata": {},
    "outputs": [
     {
@@ -877,8 +877,7 @@
     }
    ],
    "source": [
-    "fc0 = model.graph.node[2]\n",
-    "fc0w = StreamingFCLayer_Batch(fc0)\n",
+    "fc0w = getCustomOp(model.graph.node[2])\n",
     "code_gen_dir = fc0w.get_nodeattr(\"code_gen_dir_ipgen\")\n",
     "!ls {code_gen_dir}"
    ]
@@ -973,7 +972,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 60,
+   "execution_count": 38,
    "metadata": {},
    "outputs": [],
    "source": [