diff --git a/du5/program.c b/du5/program.c
index 39e0be4..7e217bb 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -127,7 +127,7 @@ int main(void) {
         printf("%s\n", cur->text);
         if (!fgets(buf, sizeof(buf), stdin)) {
             printf("Nerozumiem\n");
-            break;
+            return 0;
         }
         char *p = buf;
         while (*p && isspace((unsigned char)*p)) p++;
@@ -138,9 +138,10 @@ int main(void) {
             cur = cur->no;
         } else {
             printf("Nerozumiem\n");
-            break;
+            return 0;
         }
     }
+    printf("Koniec\n");
 
     free_tree(root);
     for (int i = 0; i < lines_count; ++i) free(lines[i]);