Fix label class bug