Fixed migration issue from previous commit