diff --git a/cv7/program.c b/cv7/program.c
index 927ff9f..4621ab9 100644
--- a/cv7/program.c
+++ b/cv7/program.c
@@ -50,13 +50,13 @@ void print_tree(struct tree* tree, int offset) {
         printf("Koniec\n");
         return;
     }
-    char r;
-    scanf("%c",&r);
+    getchar();
+    char r = getchar();
     if (r == 'a') {
         print_tree(tree->left, offset + 3);
     } else if (r == 'n') {
         print_tree(tree->right, offset + 3);
-    } else if (r == '\n') {
+    } else if (r == EOF) {
         printf("Koniec vstupu\n");
     } else {
         printf("Nerozumiem\n");