fix migration issue