Minor fix test