Fix the numpy import problem