diff --git a/content/c1/code.ipynb b/content/c1/code.ipynb
index 547c1c7..5ed91bb 100644
--- a/content/c1/code.ipynb
+++ b/content/c1/code.ipynb
@@ -83,7 +83,7 @@
    "source": [
     "sklearn_predictions = sklearn_model.predict(X_train)\n",
     "fig, ax = plt.subplots()\n",
-    "sns.scatterplot(y_train, sklearn_predictions)\n",
+    "sns.scatterplot(x=y_train, y=sklearn_predictions)\n",
     "ax.set_xlabel(r'$y$', size = 16)\n",
     "ax.set_ylabel(r'$\\hat{y}$', rotation = 0, size = 16, labelpad = 15)\n",
     "ax.set_title(r'$y$ vs. $\\hat{y}$', size = 20, pad = 10)\n",