diff --git a/pkg/probo/mitigation_service.go b/pkg/probo/mitigation_service.go index 50922fdb6..5df221d20 100644 --- a/pkg/probo/mitigation_service.go +++ b/pkg/probo/mitigation_service.go @@ -49,9 +49,10 @@ type ( ImportMitigationRequest struct { Mitigations []struct { - Name string `json:"name"` - Description string `json:"description"` - Category string `json:"category"` + Name string `json:"name"` + Description string `json:"description"` + Category string `json:"category"` + Importance coredata.MitigationImportance `json:"importance"` } `json:"mitigations"` } ) @@ -99,7 +100,7 @@ func (s MitigationService) Import( Category: mitigation.Category, State: coredata.MitigationStateNotStarted, Standards: []string{}, - Importance: coredata.MitigationImportancePreferred, + Importance: mitigation.Importance, CreatedAt: now, UpdatedAt: now, })