Fix argument reference in docs