Minor fix tests