diff --git a/cv3/program.c b/cv3/program.c
index 976fafe..7492a28 100644
--- a/cv3/program.c
+++ b/cv3/program.c
@@ -59,7 +59,7 @@ bool delit(KalkulackaZasobnik *kalkulacka) {
     if (vyber(kalkulacka, &a, &b)) {
         if (a == 0) {
             printf("division by zero\n");
-            return false; // Delenie nulou
+            return false;
         }
         return vloz(kalkulacka, b / a);
     }
@@ -77,12 +77,13 @@ int main() {
     KalkulackaZasobnik kalkulacka;
     inicializuj(&kalkulacka);
     char vstup[100];
+    bool delenieNulou = false;
 
     while (true) {
         fgets(vstup, sizeof(vstup), stdin);
 
         if (vstup[0] == '\n') {
-            break; // Prázdny vstup ukončuje cyklus
+            break;
         }
 
         float hodnota;
@@ -97,44 +98,36 @@ int main() {
                 case '+':
                     if (scitat(&kalkulacka)) {
                         zobraz_zasobnik(&kalkulacka);
-                    } else {
-                        printf("no input\n");
-                        return 0;
                     }
                     break;
                 case '-':
                     if (odcitat(&kalkulacka)) {
                         zobraz_zasobnik(&kalkulacka);
-                    } else {
-                        printf("no input\n");
-                        return 0;
                     }
                     break;
                 case '*':
                     if (nasobit(&kalkulacka)) {
                         zobraz_zasobnik(&kalkulacka);
-                    } else {
-                        printf("no input\n");
-                        return 0;
                     }
                     break;
                 case '/':
                     if (!delit(&kalkulacka)) {
-                        printf("no input\n");
-                        return 0;
+                        delenieNulou = true;
+                    } else {
+                        zobraz_zasobnik(&kalkulacka);
                     }
-                    zobraz_zasobnik(&kalkulacka);
                     break;
                 default:
-                    printf("no input\n");
                     return 0;
             }
         } else {
-            printf("no input\n");
             return 0;
         }
     }
 
-    printf("no input\n"); // Vždy vypíše "no input" na konci
+    if (!delenieNulou) {
+        printf("no input\n");
+    }
+
     return 0;
 }