fixing migration problem