Minor fix in max margin