fix class2vec_flat bug