diff --git a/package.json b/package.json
index ef3e123..08ba3fe 100644
--- a/package.json
+++ b/package.json
@@ -28,7 +28,7 @@
   "license": "Apache-2.0",
   "dependencies": {
     "@actions/core": "^1.10.0",
-    "@docker/actions-toolkit": "link:../../docker_org/actions/docker-actions-toolkit",
+    "@docker/actions-toolkit": "^0.1.0-beta.17",
     "uuid": "^9.0.0"
   },
   "devDependencies": {
diff --git a/src/main.ts b/src/main.ts
index a45e70b..1a4a9e4 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -3,9 +3,8 @@ import path from 'path';
 import * as uuid from 'uuid';
 import * as core from '@actions/core';
 import * as actionsToolkit from '@docker/actions-toolkit';
-
-import {Install} from '../../../docker_org/actions/docker-actions-toolkit/lib/docker/install';
-import {Docker} from '../../../docker_org/actions/docker-actions-toolkit/lib/docker/docker';
+import {Install} from '@docker/actions-toolkit/lib/docker/install';
+import {Docker} from '@docker/actions-toolkit/lib/docker/docker';
 
 import * as context from './context';
 import * as stateHelper from './state-helper';
diff --git a/yarn.lock b/yarn.lock
index 510549e..a58d355 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -352,9 +352,23 @@
   dependencies:
     "@cspotcode/source-map-consumer" "0.8.0"
 
-"@docker/actions-toolkit@link:../../docker_org/actions/docker-actions-toolkit":
-  version "0.0.0"
-  uid ""
+"@docker/actions-toolkit@^0.1.0-beta.17":
+  version "0.1.0-beta.17"
+  resolved "https://registry.yarnpkg.com/@docker/actions-toolkit/-/actions-toolkit-0.1.0-beta.17.tgz#71efff853f34aba59e89e13d05cf9eaf599af2a2"
+  integrity sha512-oB3qda7bTDZ3hBJxZS4TO5IK9OvelAwQicLYrfi1Uw3ROTTjpSU4sttUupY499cviLlQ/9vcTdB+8ZR+pBi1ww==
+  dependencies:
+    "@actions/core" "^1.10.0"
+    "@actions/exec" "^1.1.1"
+    "@actions/github" "^5.1.1"
+    "@actions/http-client" "^2.0.1"
+    "@actions/io" "^1.1.2"
+    "@actions/tool-cache" "^2.0.1"
+    async-retry "^1.3.3"
+    csv-parse "^5.3.5"
+    handlebars "^4.7.7"
+    jwt-decode "^3.1.2"
+    semver "^7.3.8"
+    tmp "^0.2.1"
 
 "@eslint/eslintrc@^1.2.1":
   version "1.2.1"