diff --git a/du1/program.c b/du1/program.c
index 8464358..a3d2a31 100644
--- a/du1/program.c
+++ b/du1/program.c
@@ -5,7 +5,7 @@ int main(){
 int count = 0;
 int c = 0;
 int riadok = 0;
-while(1){
+while(c != EOF){
 c = getchar();
 if(c == -1){
 break;
@@ -27,4 +27,4 @@ count = 0;
 }
 printf("\nPočet riadkov: %d\n", riadok);
 return 0;
-}
+}
\ No newline at end of file