Minor fix; missing argument