Fix imagegenerator