diff --git a/neuralNetworkTests.c b/neuralNetworkTests.c
index bad2bde..15a5c73 100644
--- a/neuralNetworkTests.c
+++ b/neuralNetworkTests.c
@@ -36,8 +36,8 @@ static void prepareNeuralNetworkFile(const char *path, const NeuralNetwork nn)
     for (int i = 0; i < nn.numberOfLayers; i++)
     {
         //const Layerlayer = &nn.layers[i];
-        int out = Layer->weights.rows;
-        int in  = Layer->weights.cols;
+        int out = nn.layers->weights.rows;
+        int in  = nn.layers->weights.cols;
 
 
         fwrite(layers->weights.buffer, sizeof(MatrixType), out * in, file);