diff --git a/main.c b/main.c
index 3ed035e9a3b502644b7dbbadf110d7df898db628..773dd3982c76d3d23aa0693e313590e1bffd84f8 100644
--- a/main.c
+++ b/main.c
@@ -25,7 +25,7 @@ int main(int argc, char const *argv[])
 
     /* on récupère le nombre de colonnes du fichier */
     fread(&c, sizeof(char), 1, f);
-    while ( c != ' ') {
+    while (c != ' ') {
         width = 10*width + strtol(&c, NULL, 10);
         fread(&c, sizeof(char), 1, f);
     }
@@ -39,9 +39,18 @@ int main(int argc, char const *argv[])
     }
     height = height/10;
 
+    /* initialisation de la matrice */
     matrix = (char **) malloc(height*sizeof(char *));
+    if (matrix == NULL) {
+        printf("Error malloc");
+        exit(1);
+    }
     for (i=0 ; i<height ; i++) {
         matrix[i] = (char *) malloc(width*sizeof(char));
+        if (matrix[i] == NULL) {
+            printf("Error malloc");
+            exit(1);
+        }
     }
 
     i = 0;