// Fix class index error