diff --git a/du6/program.c b/du6/program.c
index 998a4fb..9b7e9c7 100644
--- a/du6/program.c
+++ b/du6/program.c
@@ -99,7 +99,7 @@ int main(void){
     printf("\n%s", tree->value);
     fgets(answer, sizeof(answer), stdin);
     answer[strcspn(answer, "\n")] = '\0';
-    if(strcmp(answer, "\0") == 0){
+    if(answer[0] == '\0'){
         printf("\nKoniec vstupu\n");
         return 0;
     }
@@ -112,7 +112,6 @@ int main(void){
             return 0;
         }
         printf("%s\n", tree->value);
-         fgets(answer, sizeof(answer), stdin);
 
     }
     else if(strcmp(answer, "a") == 0){
@@ -123,7 +122,6 @@ int main(void){
             
             return 0;
         }
-         fgets(answer, sizeof(answer), stdin);
 
     }
     else{