diff --git a/u8/program.py b/u8/program.py
index ef2a401..55b3326 100644
--- a/u8/program.py
+++ b/u8/program.py
@@ -15,7 +15,11 @@ def main():
     while koren is not None:
         if koren["otazka"].endswith("?"):
             print(koren["otazka"])
-            odpoved = input().rstrip("\r")
+            try:
+                odpoved = input().rstrip("\r")
+            except EOFError:
+                print("Koniec vstupu")
+                return
             if odpoved == "a":
                 koren = koren["a"]
             elif odpoved == "n":