diff --git a/docs/source/examples/lava/nir-conversion.ipynb b/docs/source/examples/lava/nir-conversion.ipynb
index fe5a49e..5b646d5 100644
--- a/docs/source/examples/lava/nir-conversion.ipynb
+++ b/docs/source/examples/lava/nir-conversion.ipynb
@@ -32,7 +32,7 @@
     "\n",
     "ng = nir.NIRGraph(\n",
     "    nodes=[\n",
-    "        nir.Input(shape=np.array([3])),\n",
+    "        nir.Input(input_type=np.array([3])),\n",
     "        nir.Affine(weight=np.array([[8, 2, 10], [14, 3, 14]]).T, bias=np.array([1, 2])),\n",
     "        nir.LIF(\n",
     "            tau=np.array([1] * 2),\n",
@@ -40,7 +40,7 @@
     "            v_leak=np.array([0] * 2),\n",
     "            v_threshold=np.array([1] * 2),\n",
     "        ),\n",
-    "        nir.Output(shape=np.array([3])),\n",
+    "        nir.Output(input_type=np.array([3])),\n",
     "    ],\n",
     "    edges=[(0, 1), (1, 2), (2, 3)],\n",
     ")"
diff --git a/docs/source/examples/norse/nir-conversion.ipynb b/docs/source/examples/norse/nir-conversion.ipynb
index 7e48196..f4a9e32 100644
--- a/docs/source/examples/norse/nir-conversion.ipynb
+++ b/docs/source/examples/norse/nir-conversion.ipynb
@@ -25,11 +25,12 @@
     "# Create a NIR Network\n",
     "affine_weights = torch.tensor([[1.0, 2.0], [3.0, 4.0]])\n",
     "affine_bias = torch.tensor([1.0, 2.0])\n",
-    "li_tau = torch.tensor([0.9, 0.8])\n",
-    "li_r = torch.tensor([1.0, 1.0])\n",
-    "li_v_leak = torch.tensor([0.0, 0.0])\n",
+    "lif_tau = torch.tensor([0.9, 0.8])\n",
+    "lif_r = torch.tensor([1.0, 1.0])\n",
+    "lif_v_leak = torch.tensor([0.0, 0.0])\n",
+    "lif_v_th = torch.tensor([1.0, 1.0])\n",
     "nir_network = nir.NIRGraph.from_list(\n",
-    "    nir.Affine(affine_weights, affine_bias), nir.LI(li_tau, li_r, li_v_leak)\n",
+    "    nir.Affine(affine_weights, affine_bias), nir.LIF(lif_tau, lif_r, lif_v_leak, lif_v_th)\n",
     ")\n",
     "\n",
     "# Import to Norse\n",