diff --git a/du6/program.c b/du6/program.c
index ed0b7d0..efd61cc 100644
--- a/du6/program.c
+++ b/du6/program.c
@@ -83,9 +83,12 @@ void freeTree(Node *x) {
 // hlavná funkcia
 int main() {
     Node *root = readTree();          // načítame strom zo vstupu
-    printf("Expert z bufetu to vie.\n"); // vždy vypíšeme tento riadok
+     printf("Expert z bufetu to vie.\n"); // vždy vypíšeme tento riadok
 
-        
+    if (!root) {                    // ak strom je prázdny alebo chybne načítaný
+        printf("Chybna databaza\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)