Could we move this import at the beginning of the file?