small changes to migration script...