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