Fix label bug