1 files changed, 2 insertions, 1 deletions
diff --git a/test_mini/main.c b/test_mini/main.c
index 5f9175f..a0006d5 100644
--- a/test_mini/main.c
+++ b/test_mini/main.c
@@ -7,7 +7,8 @@ int         main(int argc, char **argv)
     int i = 0;
     char *input;
 
-    input = malloc(sizeof(char) * ft_strlen(argv[1]) + 2);
+    if (!(input = malloc(sizeof(char) * ft_strlen(argv[1]) + 2)))
+        return(0);
     ft_strlcpy(input, argv[1], ft_strlen(argv[1]) + 1);
     i = ft_strlen(input);
     input[i + 1] = '\0';