diff --git a/du1/program.c b/du1/program.c
index 99b6bc5..4309cc7 100644
--- a/du1/program.c
+++ b/du1/program.c
@@ -13,8 +13,7 @@ int main() {
     char line[SIZE];
     char* end;
 
-    char* r = fgets(line,SIZE,stdin);
-    if (r == NULL){
+    if (fgets(line,SIZE,stdin) == NULL){
         printf("Chyba: nacitanie sa nepodarilo.\n");
         return 1;
 }
@@ -25,7 +24,7 @@ int main() {
         return 1;
 }
 
-while(r != NULL){
+while(fgets(line,SIZE,stdin) != NULL){
     if(line[0] == '\n'){
         break;
     }
@@ -43,4 +42,5 @@ for(int i = 1; i < k; i++){
     result = result * x + field[i];
 }
    printf("Vysledok je: %.2f\n", result);
+   return 0;
 }