Corrected migration issue