diff --git a/main.cpp b/main.cpp
index 14d1577..8397089 100644
--- a/main.cpp
+++ b/main.cpp
@@ -141,7 +141,7 @@ float sigmoidDerivative(float x) {
 int main() {
     NeuralNetwork nn;
     nn.addLayer(2, 2, sigmoid, sigmoidDerivative);
-    nn.addLayer(1, 1, sigmoid, sigmoidDerivative);
+    nn.addLayer(1, 2, sigmoid, sigmoidDerivative);
 
     std::vector<std::vector<float>> ref = {
         {0, 0},