Fix tests after args were added