Fix import math