Fix up migration