diff --git a/du6/program.c b/du6/program.c
index 7c7e282..c8999e9 100644
--- a/du6/program.c
+++ b/du6/program.c
@@ -93,7 +93,7 @@ int main(void){
     int numOfLeaves = count_leaves(tree);
     char answer[SIZE];
     printf("Expert z bufetu to vie.\n");
-    if(!tree){
+    if(!tree || !whitespace){
         printf("Chybna databaza\n");
         return 0;
     }