diff --git a/du5/program.c b/du5/program.c
index 75ac44c..47e5946 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -111,11 +111,12 @@ int main() {
         return 0;
     }
 
-    if (!EmptyLine()) {
+    char c = getchar();
+    if (c != '\n' && c != EOF) {
         printf("Chyba nacitania\n");
         freeTree(root);
         return 0;
-    }
+}
 
     printf("Pozna %d druhov ovocia a zeleniny.\n", Leaves(root));
     printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");