Minor fix and doc