Fix label regression