diff --git a/du5/program.c b/du5/program.c
index b9eb1a6..d01014b 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -48,20 +48,18 @@ void spustac(struct uzol* uzol) {
     }
     
     char odpoved; // iacitavam a/n
-    scanf(" %c",&odpoved);
+    if (scanf(" %c",&odpoved) != 1) { // ak mi nenacita hodnotu
+        printf ("Koniec vstupu\n");
+        return; 
+    }
 
     if (odpoved == 'a') {
         spustac (uzol -> lavy);
     } else if (odpoved == 'n') {
         spustac (uzol -> pravy);
-    } else if (odpoved == '\n'){ 
-        printf ("Koniec vstupu\n");
     } else { 
         printf ("Nerozumiem\n");
     }
-
-
-
 }
 
 void destroy_tree (struct uzol* strom){