diff --git a/du6/program.c b/du6/program.c
index a9c0dc7..80436d8 100644
--- a/du6/program.c
+++ b/du6/program.c
@@ -93,6 +93,10 @@ int main(void){
     int numOfLeaves = count_leaves(tree);
     char answer[SIZE];
     printf("Expert z bufetu to vie.\n");
+    if(!tree){
+        printf("Chybna databaza\n");
+        return 0;
+    }
     printf("Pozna %d druhov ovocia a zeleniny.\n", numOfLeaves);
     printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.");
     while(tree){