Bug fix

math error