Minor fix, missing raise