diff --git a/du5/program.c b/du5/program.c
index 16ec504..14b0599 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -131,7 +131,7 @@ int main()
     int root = buildATree();
 
     printf("Expert z bufetu to vie.\n");
-    if (answerCount == 0 || root == -1 || nodeCount == 0)
+    if (indexLine != lineCount || root == -1 || nodeCount == 0)
     {
         printf("Chybna databaza\n");
     }