Add check for target image mode