From ba8d982e5d7f67e677c2995355ea3b97d9051b22 Mon Sep 17 00:00:00 2001
From: Illia Korpan <illia.korpan@student.tuke.sk>
Date: Wed, 15 Apr 2026 09:41:06 +0200
Subject: [PATCH] 123

---
 du5/program.c | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/du5/program.c b/du5/program.c
index 312aa4e..0908d70 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -119,9 +119,11 @@ int main(void) {
 
     Node *cur = root;
     char buf[128];
+    int valid_input = 0;
     while (cur) {
         if (cur->is_answer) {
             printf("*%s\n", cur->text);
+            valid_input = 1;
             break;
         }
         printf("%s\n", cur->text);
@@ -138,17 +140,19 @@ int main(void) {
         }
         if (c == 'a' || c == 'A') {
             cur = cur->yes;
-            printf("Koniec\n");
-            break;
+            valid_input = 1;
         } else if (c == 'n' || c == 'N') {
             cur = cur->no;
-            printf("Koniec\n");
-            break;
+            valid_input = 1;
         } else {
             printf("Nerozumiem\n");
             break;
         }
     }
+    
+    if (valid_input) {
+        printf("Koniec\n");
+    }
 
     free_tree(root);
     for (int i = 0; i < lines_count; ++i) free(lines[i]);