diff --git a/u8/program.py b/u8/program.py
index 55b3326..1648a46 100644
--- a/u8/program.py
+++ b/u8/program.py
@@ -57,10 +57,13 @@ def pocet_odpovedi(tree_lines):
 
 def read_tree_lines():
     tree_lines = []
-    text = input().rstrip("\r")
-    while text:
-        tree_lines.append(text)
+    try:
         text = input().rstrip("\r")
+        while text:
+            tree_lines.append(text)
+            text = input().rstrip("\r")
+    except EOFError:
+        pass
     return tree_lines
 
 def controla_batabaz(tree_lines):