Bug fix: use image name instead of image object