diff --git a/cv8/program.c b/cv8/program.c
index c82a6cb..c620766 100644
--- a/cv8/program.c
+++ b/cv8/program.c
@@ -43,11 +43,11 @@ void search(struct tree* this){
     if(fgets(buffer, 5, stdin) == NULL){
         printf("Koniec vstupu\n");
         return;
-    }  
-    //char* r = fgets(buffer,5,stdin);
-    //int x = strlen(buffer);
-    //buffer[x-1]='\0';
-   
+    }   
+    if(buffer[0] == '\n'){
+        printf("Koniec vstupu\n");
+        return;
+    }
         if(buffer[0] == '\n'){
             char* r = fgets(buffer,5,stdin);
         }