minor change to migration script