diff --git a/cv7/program.c b/cv7/program.c
index 6f45b80..7fd12c5 100644
--- a/cv7/program.c
+++ b/cv7/program.c
@@ -83,14 +83,14 @@ struct tree* read_tree() {
     char* r = fgets(buffer, SIZE, stdin);
     if (r == NULL) {
        // fprintf(stderr, "Chybna databaza\n");
-        exit(1);  // Exit the program if reading fails
+        node = NULL;  // Exit the program if reading fails
     }
 
     // Allocate memory for a new tree node
     struct tree* node = calloc(1, sizeof(struct tree));
     if (!node) {
-        printf("Memory allocation failed!\n");
-        exit(1);
+        //printf("Memory allocation failed!\n");
+        node = NULL;
     }
 
     // Copy the input line into the node's value