diff --git a/cloud/agent/install.go b/cloud/agent/install.go
index 2ba9859..e212d9a 100644
--- a/cloud/agent/install.go
+++ b/cloud/agent/install.go
@@ -112,6 +112,7 @@
 			System: 4096,
 			Data:   128,
 		},
+		Architecture:   osImage.Config.ProductInfo.Architecture(),
 		SystemImage:    systemImage,
 		EFIPayload:     efiPayload,
 		ABLoader:       structfs.Bytes(abloader),