Minor fix affects test cases