Small fix migration