fix label finding bug