Update schema so the correction tests pass