diff --git a/cv7/program.c b/cv7/program.c
index 0eb78fb..937af61 100644
--- a/cv7/program.c
+++ b/cv7/program.c
@@ -14,10 +14,10 @@ struct tree* read_tree(int* counter) {
     char buffer[SIZE];
     memset(buffer, 0, SIZE);
     char* r = fgets(buffer, SIZE, stdin);
-    if (buffer[0] == '\n'){
+    if(r == NULL){
         printf("Expert z bufetu to vie.\n");
         printf("Chybna databaza\n");
-        return 0;
+        return NULL;
     }
     if (buffer[0] == '\n') {
         return NULL;