Fix migration to match create table