diff --git a/du3/program.c b/du3/program.c
index 83279e4..194b66e 100644
--- a/du3/program.c
+++ b/du3/program.c
@@ -68,12 +68,12 @@ int main(){
             case '-': skutocny_vysledok = cislo1 - cislo2; break;
             case '*': skutocny_vysledok = cislo1 * cislo2; break;
             case '/':
-                if (cislo2 == 0) {
-                    chyba = 1;
+               if (cislo2 == 0){
+                  skutocny_vysledok = INFINITY;
                 }
                 else
                 {
-                    skutocny_vysledok = cislo1 / cislo2;
+                  skutocny_vysledok = cislo1 / cislo2;
                 }
                 break;
             default: chyba = 1;