fix missing math import