fix missing import in image.py