Fix corpus shape