Fixed migration issue