diff --git a/a1/program.c b/a1/program.c
index f7b9267..71ee50a 100644
--- a/a1/program.c
+++ b/a1/program.c
@@ -4,6 +4,7 @@
 int main()
 {
     int pocetRiadkov = 0;
+    int poslednyChar = EOF;
     while (1)
     {
         int vstup = getchar();
@@ -31,8 +32,14 @@ int main()
         {
             putchar(vstup);
         }
+        poslednyChar = vstup;
 
     }
+    if (poslednyChar != '\n' && poslednyChar != EOF)
+    {
+        putchar('\n');
+        pocetRiadkov += 1;
+    }
     printf("Lines_count: %d\n", pocetRiadkov);
     return 0;
 }