diff --git a/neuralNetwork.c b/neuralNetwork.c
index bd8f164..5f0e4dd 100644
--- a/neuralNetwork.c
+++ b/neuralNetwork.c
@@ -3,6 +3,7 @@
 #include <math.h>
 #include <string.h>
 #include "neuralNetwork.h"
+#include "matrix.h"
 
 #define BUFFER_SIZE 100
 #define FILE_HEADER_STRING "__info2_neural_network_file_format__"
diff --git a/neuralNetworkTests.c b/neuralNetworkTests.c
index e10b18a..73950f8 100644
--- a/neuralNetworkTests.c
+++ b/neuralNetworkTests.c
@@ -4,8 +4,11 @@
 #include <math.h>
 #include "unity.h"
 #include "neuralNetwork.h"
+#include "matrix.h"
 
 
+#define NN_HEADER "__info2_neural_network_file_format__\n"
+
 //das auch zu machen
 static void writeMatrix(FILE *file, const Matrix *m) {
     unsigned short rows = (unsigned short)m->rows;