Removing unneeded imports in migration script