Updated migration script according to #962