diff --git a/Assignments/Assignment1/linear_regression.ipynb b/Assignments/Assignment1/linear_regression.ipynb
index 08087b3..0cef8e8 100644
--- a/Assignments/Assignment1/linear_regression.ipynb
+++ b/Assignments/Assignment1/linear_regression.ipynb
@@ -730,7 +730,7 @@
     "    J_vals = np.zeros((len(theta0_vals), len(theta1_vals)), dtype=np.float64)\n",
     "    for i, v0 in enumerate(theta0_vals):\n",
     "        for j, v1 in enumerate(theta1_vals):\n",
-    "            theta = np.array((theta0_vals[i], theta1_vals[j])).reshape(2, 1)\n",
+    "            theta = np.array((v0, v1)).reshape(2, 1)\n",
     "            J_vals[i, j] = compute_cost(X, y, theta)\n",
     "\n",
     "    T0, T1 = np.meshgrid(theta0_vals, theta1_vals)\n",