Fix migration snafu