diff --git a/du5/program.c b/du5/program.c
index d5e249c..12cd23b 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -41,10 +41,7 @@ int zrataj_produkty(struct uzol* uzol) {
 void spustac(struct uzol* uzol) {
     if (uzol == NULL) return; // ak nemam strom koncim
     printf ("%s",uzol -> otazka); // vypisem content textu struktury
-    if (uzol -> pravy == NULL && uzol -> lavy == NULL) { // ak nema pokracovanie, napisem ze nema pokracovanie
-        printf ("Koniec\n");
-        return;
-    }
+
     char odpoved; // iacitavam a/n
     scanf(" %c",&odpoved);
 
@@ -54,6 +51,11 @@ void spustac(struct uzol* uzol) {
         spustac (uzol -> pravy);
     } // uvidim co test ocakava ked dam zle odpoved
 
+    if (uzol -> pravy == NULL && uzol -> lavy == NULL) { // ak nema pokracovanie, napisem ze nema pokracovanie
+        printf ("Koniec\n");
+        return;
+    }
+
 }
 
 void destroy_tree (struct uzol* strom){