diff --git a/exercises/navis/navis_exercise_2.ipynb b/exercises/navis/navis_exercise_2.ipynb
index b3e6b5e..93d0705 100644
--- a/exercises/navis/navis_exercise_2.ipynb
+++ b/exercises/navis/navis_exercise_2.ipynb
@@ -301,7 +301,7 @@
     "        The matplotlib figure and axis.\n",
     "\n",
     "    \"\"\"\n",
-    "    to_plot = nl[nl.t_type == \"Lamp5 Ntn1 Npy2r\"]\n",
+    "    to_plot = nl[nl.t_type == t_type]\n",
     "\n",
     "    # Offset the neurons slightly along the x-axis\n",
     "    to_plot = [n + [offset * i, 0, 0] for i, n in enumerate(to_plot)]\n",