diff --git a/du4/program.c b/du4/program.c
index a669d39..2d7b0be 100644
--- a/du4/program.c
+++ b/du4/program.c
@@ -38,8 +38,8 @@ int main() {
     while (fgets(line, SIZE, stdin) != NULL) {
         char *end = NULL;
         int votes = strtol(line, &end, 10);
-        if (votes == 0) {
-            continue; // Neplatný vstup, preskočiť
+        if (end == line || *end == '\0' || votes <= 0) {
+            break; // Neplatný vstup, ukončiť načítanie
         }
         
         // Preskočiť medzeru po čísle hlasov