Fixed ZeroDivisionError -- stupid me.