diff --git a/cv7/program.c b/cv7/program.c
index 94e7a6b..55a43e9 100644
--- a/cv7/program.c
+++ b/cv7/program.c
@@ -52,12 +52,13 @@ void print_tree(struct tree* tree,int offset){
     }
     getchar();
     char r = getchar();
-    getchar();
     if (r == 'a') {
         print_tree(tree->left, offset + 3);
     } else if (r == 'n') {
         print_tree(tree->right, offset + 3);
-    } else {
+    } else if (r == '\n'){
+        printf("Koniec vstupu\n");
+    }else {
         printf("Nerozumiem\n");
         exit(0);
     }