Fix error in math