Fix to migration