diff --git a/src/main.ts b/src/main.ts
index 9671871..5b53745 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -21,8 +21,11 @@ actionsToolkit.run(
 
     const install = new Install({
       runDir: runDir,
-      version: input.version,
-      channel: input.channel || 'stable',
+      source: {
+        type: 'archive',
+        version: input.version,
+        channel: input.channel || 'stable'
+      },
       contextName: input.context || 'setup-docker-action',
       daemonConfig: input.daemonConfig
     });