[CHANGE]modify migration to work