fix mistake

Author: w0ng952

pkg/templates/deployment.go

@@ -69,6 +69,8 @@ func (a *PrometheusRuleTemplateManager) CreateFromDeployment(deployment *apps.De
 
 	prometheusRules := map[string]*monitoringv1.PrometheusRule{}
 	annotations := params.Deployment.GetAnnotations()
+	params.CustomLabels = extractCustomLabels(annotations)
+
 	for k, v := range annotations {
 		if !strings.HasPrefix(k, heimPrefix) {
 			continue
@@ -79,8 +81,6 @@ func (a *PrometheusRuleTemplateManager) CreateFromDeployment(deployment *apps.De
 			continue
 		}
 
-		params.CustomLabels = extractCustomLabels(annotations)
-
 		templateName := strings.TrimLeft(k, fmt.Sprintf("%s/", heimPrefix))
 		logger.Infow("template selected", "template", templateName)
 		template, ok := a.templates[templateName]

pkg/templates/ingress.go

@@ -51,6 +51,7 @@ func (a *PrometheusRuleTemplateManager) CreateFromIngress(ingress *networkingv1.
 
 	prometheusRules := map[string]*monitoringv1.PrometheusRule{}
 	annotations := ingress.GetAnnotations()
+	params.CustomLabels = extractCustomLabels(annotations)
 
 	for k, v := range annotations {
 		if !strings.HasPrefix(k, heimPrefix) {
@@ -62,8 +63,6 @@ func (a *PrometheusRuleTemplateManager) CreateFromIngress(ingress *networkingv1.
 			continue
 		}
 
-		params.CustomLabels = extractCustomLabels(annotations)
-
 		templateName := strings.TrimLeft(k, fmt.Sprintf("%s/", heimPrefix))
 		template, ok := a.templates[templateName]
 		if !ok {