diff --git a/du5/program.c b/du5/program.c
index 7c14ecf..b7829ad 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -33,7 +33,7 @@ int main(){
         if (value == 0&&line[0]=='\n'){
             // Premena sa nepodarila
             break;
-        }else if(value==0&&line[0]!='\n'&&size==0){
+        }else if((value==0&&line[0]!='\n'&&size==0)||(size==0&&line==EOF)){
 	    printf("Nepodarilo nacitat nic\n");
 	    return 0;
 	}