diff --git a/du5/program.c b/du5/program.c
index 5fcc329..f74119c 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -8,9 +8,8 @@ int main()
     char line[SIZE];
     char answer[SIZE];
 
-    while (1)
+    while (fgets(line, SIZE, stdin) != NULL)
     {
-        char* p = fgets(line, SIZE, stdin);
         if (line[0] == NULL)
         {
             break;