From f1e83deb109bb8a6842a2eb30c4a9940b0546e60 Mon Sep 17 00:00:00 2001
From: Bryce Soghigian <49734722+Bryce-Soghigian@users.noreply.github.com>
Date: Sat, 7 Dec 2024 00:53:43 +0000
Subject: [PATCH] fix: extending key

---
 pkg/providers/imagefamily/image.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/pkg/providers/imagefamily/image.go b/pkg/providers/imagefamily/image.go
index d03893efe..c9f03cde1 100644
--- a/pkg/providers/imagefamily/image.go
+++ b/pkg/providers/imagefamily/image.go
@@ -108,9 +108,9 @@ func (p *Provider) KubeServerVersion(ctx context.Context) (string, error) {
 }
 
 func (p *Provider) getImageIDSIG(ctx context.Context, imgStub DefaultImageOutput) (string, error) {
-	// Note: one could argue that we could narrow the key one level further to ImageDefinition since no two AKS ImageDefinitions that are supported 
-	// by karpenter have the same name, but for EdgeZone support this is not the case. 
-	key := fmt.Sprintf(sharedImageKey, imgStub.GalleryName, imgStub.ImageDefinition) 
+	// Note: one could argue that we could narrow the key one level further to ImageDefinition since no two AKS ImageDefinitions that are supported
+	// by karpenter have the same name, but for EdgeZone support this is not the case.
+	key := fmt.Sprintf(sharedImageKey, imgStub.GalleryName, imgStub.ImageDefinition)
 	if imageID, ok := p.imageCache.Get(key); ok {
 		return imageID.(string), nil
 	}
@@ -130,7 +130,7 @@ func (p *Provider) getImageIDSIG(ctx context.Context, imgStub DefaultImageOutput
 }
 
 func (p *Provider) getImageIDCIG(publicGalleryURL, communityImageName string) (string, error) {
-	key := fmt.Sprintf(communityImageName)
+	key := fmt.Sprintf(communityImageKey, publicGalleryURL, communityImageName)
 	if imageID, ok := p.imageCache.Get(key); ok {
 		return imageID.(string), nil
 	}