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