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