diff --git a/du5/program.c b/du5/program.c
index fe4c38c..93aa984 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -72,6 +72,15 @@ int main(void) {
         printf("Chybna databaza\n");
         return 0;
     }
+    char sep[SIZE];
+    if (fgets(sep, SIZE, stdin) == NULL ||
+           (sep[0] != '\n' && sep[0] != '\r' && sep[0] != '\0')) {
+        printf("Expert z bufetu to vie.\n");
+        printf("Chybna databaza\n");
+        destroy_tree(root);
+        return 0;
+           }
+
     int leaves = count_leaves(root);
     printf("Expert z bufetu to vie.\n");
     printf("Pozna %d druhov ovocia a zeleniny.\n", leaves);