diff --git a/du1/program.c b/du1/program.c
index 5fe819a..914e228 100644
--- a/du1/program.c
+++ b/du1/program.c
@@ -25,7 +25,7 @@ int main() {
         return 1;
 }
 
-while(r != NULL){
+while (fgets(line, SIZE, stdin) != NULL) {
     if(line[0] == '\n'){
         break;
     }