diff --git a/du6/program.c b/du6/program.c
index 6dd63e7..9ff4562 100644
--- a/du6/program.c
+++ b/du6/program.c
@@ -88,6 +88,11 @@ void freeTree(Node *x) {
 int main() {
     Node *root = readTree();          // načítame strom zo vstupu
 
+    if (!root) {                    // ak strom je prázdny
+        printf("Koniec vstupu\n");
+        return 0;
+    }
+
     fgets(s, MAX, stdin);             // prečítame prázdny riadok medzi databázou a vstupom používateľa
 
     int k = count(root);              // spočítame počet listov (druhov tovaru)