fixed multi-class failure bug