Minor fix on the interface