Fixed migration problem