Minor fix in api