diff --git a/pkg/container/docker_run.go b/pkg/container/docker_run.go
index df296e0..65a373f 100644
--- a/pkg/container/docker_run.go
+++ b/pkg/container/docker_run.go
@@ -240,8 +240,8 @@ func RunnerArch(ctx context.Context) string {
 
 	archMapper := map[string]string{
 		"x86_64":  "X64",
-		"386":     "x86",
-		"aarch64": "arm64",
+		"386":     "X86",
+		"aarch64": "ARM64",
 	}
 	if arch, ok := archMapper[info.Architecture]; ok {
 		return arch
diff --git a/pkg/container/host_environment.go b/pkg/container/host_environment.go
index e0a18c7..c6a2228 100644
--- a/pkg/container/host_environment.go
+++ b/pkg/container/host_environment.go
@@ -387,11 +387,13 @@ func (*HostEnvironment) JoinPathVariable(paths ...string) string {
 	return strings.Join(paths, string(filepath.ListSeparator))
 }
 
+// Reference for Arch values for runner.arch
+// https://docs.github.com/en/actions/learn-github-actions/contexts#runner-context
 func goArchToActionArch(arch string) string {
 	archMapper := map[string]string{
 		"x86_64":  "X64",
-		"386":     "x86",
-		"aarch64": "arm64",
+		"386":     "X86",
+		"aarch64": "ARM64",
 	}
 	if arch, ok := archMapper[arch]; ok {
 		return arch