Minor fix in assert