Fix the maths problem