FIX in ImageModelling module