Add additional arg to step def