Update keyword arguments in tests