diff --git a/cv8/program.c b/cv8/program.c
index dc280f1..0c4577a 100644
--- a/cv8/program.c
+++ b/cv8/program.c
@@ -25,6 +25,11 @@ struct tree* read_tree(int* counter) {
     char* r = fgets(buffer, SIZE, stdin);
     assert(r);
 
+    if(r == NULL){
+        printf("Koniec vstupu\n");
+        exit(0);
+    }
+
     struct tree* node = calloc(1, sizeof(struct tree));
     memcpy(node->v, buffer, SIZE);