diff --git a/a1/program.c b/a1/program.c
index c8e5cee..bd716e1 100644
--- a/a1/program.c
+++ b/a1/program.c
@@ -4,14 +4,11 @@
 int main()
 {
     int pocetRiadkov = 0;
-    while (1)
+    while (vstup)
     {
         int vstup = getchar();
-        if (vstup == EOF)
-        {
-            break;
-        }
-
+        if (vstup == EOL) {break;}
+        
         if (vstup == '\n')
         {
             pocetRiadkov += 1;