Fix Label in Imagination