fix up migration problem