diff --git a/mpl-xor/main.py b/mpl-xor/main.py
index 99ccf65c35af767372327c9606480fabb671e359..82a28860c61daeffcd66496ff14bdf5fb7e969b2 100644
--- a/mpl-xor/main.py
+++ b/mpl-xor/main.py
@@ -16,7 +16,7 @@ class MultilayerPerceptron(nn.Module):
         return self.layers(x)
 
 
-model = MultilayerPerceptron(5)
+model = MultilayerPerceptron(3)
 model.cuda()
 
 loss_function = nn.MSELoss()