fix math function bug