diff --git a/08_2_dataset_loade_logistic.py b/08_2_dataset_loade_logistic.py
index 43ba9f4..c027c96 100644
--- a/08_2_dataset_loade_logistic.py
+++ b/08_2_dataset_loade_logistic.py
@@ -41,7 +41,7 @@ def __init__(self):
         super(Model, self).__init__()
         self.l1 = torch.nn.Linear(8, 6)
         self.l2 = torch.nn.Linear(6, 4)
-        self.l3 = torch.nn.Linear(4, 1)
+        self.l3 = torch.nn.Linear(4, 2)
 
         self.sigmoid = torch.nn.Sigmoid()