Fix wrong argument name in tests