diff --git a/action.cue b/action.cue
index eed949e..c2fe1a3 100644
--- a/action.cue
+++ b/action.cue
@@ -1,8 +1,9 @@
 package action
 
-action: {
+import "json.schemastore.org/github"
+
+action: github.#Action & {
 	name:        "Setup CUE environment"
-	icon:        "terminal"
 	description: "Setup a Go environment and add it to the PATH."
 	inputs: version: {
 		description: #"The CUE version to setup. Must be a valid semantic version string like "v0.4.0" or "latest""#
@@ -10,7 +11,10 @@ action: {
 		default:     "latest"
 	}
 	outputs: "cue-path": description: "Path to the cached CUE binary"
-	branding: color: "blue"
+	branding: {
+		icon:  "terminal"
+		color: "blue"
+	}
 	runs: {
 		using: "node12"
 		main:  "dist/index.js"
diff --git a/action.yml b/action.yml
index ba9cac6..ca25246 100644
--- a/action.yml
+++ b/action.yml
@@ -1,7 +1,6 @@
 # Generated by cue cmd genaction; do not edit
 
 name: Setup CUE environment
-icon: terminal
 description: Setup a Go environment and add it to the PATH.
 inputs:
   version:
@@ -12,6 +11,7 @@ outputs:
   cue-path:
     description: Path to the cached CUE binary
 branding:
+  icon: terminal
   color: blue
 runs:
   using: node12