minor: fixing import
build: refactoring migration script