diff --git a/assignment2/regression.ipynb b/assignment2/regression.ipynb
index c4eaff1..ebc6bd6 100644
--- a/assignment2/regression.ipynb
+++ b/assignment2/regression.ipynb
@@ -625,10 +625,10 @@
         "                         (\"linear_regression\", linear_regression)])\n",
         "    pipeline.fit(X[:, np.newaxis], y)\n",
         "\n",
-        "    X_test = np.linspace(0, 1, 100)\n",
-        "    plt.plot(X_test, pipeline.predict(X_test[:, np.newaxis]),\n",
+        "    X_test_i = np.linspace(0, 1, 100)\n",
+        "    plt.plot(X_test_i, pipeline.predict(X_test_i[:, np.newaxis]),\n",
         "             label=\"Model\")\n",
-        "    plt.plot(X_test, true_fun(X_test), label=\"True function\")\n",
+        "    plt.plot(X_test_i, true_fun(X_test_i), label=\"True function\")\n",
         "    plt.scatter(X, y, edgecolor='b', s=20, label=\"Samples\")\n",
         "    plt.xlabel(\"x\")\n",
         "    plt.ylabel(\"y\")\n",
@@ -806,4 +806,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}