diff --git a/models/babyunet.py b/models/babyunet.py
index 6115bd9..f2f5b9a 100644
--- a/models/babyunet.py
+++ b/models/babyunet.py
@@ -29,7 +29,7 @@ def forward(self, x):
         x = self.pool1(c1)
         c2 = self.conv2(x)
         x = self.pool2(c2)
-        self.conv3(x)
+        x = self.conv3(x)
 
         x = self.up1(x)
         x = torch.cat([x, c2], 1)