Fixing argument rename

This is wrong because both @nilsbore and me made PRs in parallel, so he didn't see my changes and I didn't see his.