ZeroDivisionError fixed