Fixed migration errors