diff --git a/du5/program.c b/du5/program.c
index cf02ca8..71d3081 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -120,7 +120,7 @@ int main()
     //zisk. korena
     int root = buildATree();
 
-    if (root == -1)
+    if (tree[root].yes == -1 && tree[root].no == -1)
     {
         printf("Koniec vstupu.\n");
         return 0;