fix migration step argument integration