diff --git a/management/server/policy.go b/management/server/policy.go index f889c60f2..4a64ddc91 100644 --- a/management/server/policy.go +++ b/management/server/policy.go @@ -159,11 +159,12 @@ type Policy struct { // Copy returns a copy of the policy. func (p *Policy) Copy() *Policy { c := &Policy{ - ID: p.ID, - Name: p.Name, - Description: p.Description, - Enabled: p.Enabled, - Rules: make([]*PolicyRule, len(p.Rules)), + ID: p.ID, + Name: p.Name, + Description: p.Description, + Enabled: p.Enabled, + Rules: make([]*PolicyRule, len(p.Rules)), + PostureCheck: p.PostureCheck, } for i, r := range p.Rules { c.Rules[i] = r.Copy() diff --git a/management/server/sqlite_store.go b/management/server/sqlite_store.go index 1bc2db3f1..3b6a8c7e9 100644 --- a/management/server/sqlite_store.go +++ b/management/server/sqlite_store.go @@ -15,6 +15,7 @@ import ( nbdns "github.com/netbirdio/netbird/dns" "github.com/netbirdio/netbird/management/server/account" + "github.com/netbirdio/netbird/management/server/checks" nbpeer "github.com/netbirdio/netbird/management/server/peer" "github.com/netbirdio/netbird/management/server/status" "github.com/netbirdio/netbird/management/server/telemetry" @@ -63,7 +64,7 @@ func NewSqliteStore(dataDir string, metrics telemetry.AppMetrics) (*SqliteStore, err = db.AutoMigrate( &SetupKey{}, &nbpeer.Peer{}, &User{}, &PersonalAccessToken{}, &Group{}, &Rule{}, &Account{}, &Policy{}, &PolicyRule{}, &route.Route{}, &nbdns.NameServerGroup{}, - &installation{}, &account.ExtraSettings{}, + &installation{}, &account.ExtraSettings{}, &checks.PostureCheck{}, ) if err != nil { return nil, err