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