Fix multiclass container
0.6.2