diff --git a/Examples_from_manuscript.ipynb b/Examples_from_manuscript.ipynb
index 2c5cca616959893bd4d5ed43b577f29c419ccaf2..5c89cd9a47ebefb564c3ffced7c7af97366e4477 100644
--- a/Examples_from_manuscript.ipynb
+++ b/Examples_from_manuscript.ipynb
@@ -118,10 +118,10 @@
     "    A customised model with a 1x1 Conv, and 2 Dense layers (16, \n",
     "    output shape). Dropout is used after Conv layer.\n",
     "    \"\"\"\n",
-    "    def __init__(self, shape_inputs: list, shape_outputs: list):\n",
+    "    def __init__(self, input_shape: list, output_shape: list):\n",
     "    \n",
     "        # set attributes shape_inputs and shape_outputs\n",
-    "        super().__init__(shape_inputs[0], shape_outputs[0])\n",
+    "        super().__init__(input_shape[0], output_shape[0])\n",
     "\n",
     "        # apply to model\n",
     "        self.set_model()\n",