diff --git a/8_sgd_vs_gd/gd_and_sgd.ipynb b/8_sgd_vs_gd/gd_and_sgd.ipynb
index 6064565..5c293fd 100644
--- a/8_sgd_vs_gd/gd_and_sgd.ipynb
+++ b/8_sgd_vs_gd/gd_and_sgd.ipynb
@@ -535,7 +535,7 @@
     "        \n",
     "    return w, b, cost, cost_list, epoch_list\n",
     "\n",
-    "w_sgd, b_sgd, cost_sgd, cost_list_sgd, epoch_list_sgd = SGD(scaled_X,scaled_y.reshape(scaled_y.shape[0],),10000)\n",
+    "w_sgd, b_sgd, cost_sgd, cost_list_sgd, epoch_list_sgd = stochastic_gradient_descent(scaled_X,scaled_y.reshape(scaled_y.shape[0],),10000)\n",
     "w_sgd, b_sgd, cost_sgd"
    ]
   },
@@ -704,4 +704,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
\ No newline at end of file
+}