Minor argument fix