fix panic

elastic · Feb 24, 2023 · 30ce399 · 30ce399
1 parent bdcb6a1
commit 30ce399
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pkg/client/unit.go b/pkg/client/unit.go
@@ -331,18 +331,22 @@ func newUnit(
 	cfgIdx uint64,
 	client *clientV2) *Unit {
 
-	return &Unit{
+	unit := Unit{
 		id:            id,
 		unitType:      unitType,
 		config:        cfg,
 		configIdx:     cfgIdx,
 		expectedState: exp,
-		features:      client.features,
 		logLevel:      logLevel,
 		state:         UnitStateStarting,
 		stateMsg:      "Starting",
 		client:        client,
 		actions:       make(map[string]Action),
 		diagHooks:     make(map[string]diagHook),
 	}
+
+	if client != nil {
+		unit.features = client.features
+	}
+	return &unit
 }