diff --git a/autodiff_puzzlers.ipynb b/autodiff_puzzlers.ipynb
index 15ab52f..b0d9c95 100644
--- a/autodiff_puzzlers.ipynb
+++ b/autodiff_puzzlers.ipynb
@@ -4780,7 +4780,7 @@
         "\n",
         "\n",
         "def f(x: Shaped[Array, \"I\"], y: Shaped[Array, \"W\"]) -> Shaped[Array, \"O\"]:\n",
-        "    return np.convolve(x, y, mode=\"valid\") / W\n",
+        "    return np.convolve(x, np.flip(y), mode=\"valid\") / W\n",
         "\n",
         "\n",
         "def jac(\n",
@@ -7627,4 +7627,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 5
-}
\ No newline at end of file
+}