Use ARM predicate for bottlerocket SSM parameter

Signed-off-by: Jacob Vallejo <[email protected]>
laurovenancio · Aug 31, 2020 · d9a2b41 · d9a2b41
1 parent 8cba8e2
commit d9a2b41
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pkg/ami/ssm_resolver.go b/pkg/ami/ssm_resolver.go
@@ -2,7 +2,6 @@ package ami
 
 import (
 	"fmt"
-	"strings"
 
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/service/eks"
@@ -88,8 +87,7 @@ func MakeManagedSSMParameterName(version, imageFamily, amiType string) (string,
 // instanceEC2ArchName returns the name of the architecture as used by EC2
 // resources.
 func instanceEC2ArchName(instanceType string) string {
-	// eg: a1.large - an ARM instance type.
-	if strings.HasPrefix(instanceType, "a") {
+	if utils.IsARMInstanceType(instanceType) {
 		return "arm64"
 	}
 	return "x86_64"