diff --git a/FFNN Diabetes.ipynb b/FFNN Diabetes.ipynb
index 4fd014b..4948a13 100644
--- a/FFNN Diabetes.ipynb	
+++ b/FFNN Diabetes.ipynb	
@@ -236,7 +236,7 @@
     "#size_average = True --> the losses are averaged over observations for each minibatch\n",
     "criterion = torch.nn.BCELoss(size_average=True)   \n",
     "# We will use SGD with momentum with a learning rate of 0.1\n",
-    "optimizer = torch.optim.SGD(model.parameters(), lr=0.1, momentum=0.9)"
+    "optimizer = torch.optim.SGD(net.parameters(), lr=0.1, momentum=0.9)"
    ]
   },
   {
@@ -300,4 +300,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}
\ No newline at end of file