Could you please move this import up, and put it close to the numpy import.