Fix wax-math