diff --git a/__tests__/context.test.ts b/__tests__/context.test.ts
index 2f42259..f590410 100644
--- a/__tests__/context.test.ts
+++ b/__tests__/context.test.ts
@@ -10,7 +10,7 @@ import {Docker} from '@docker/actions-toolkit/lib/docker/docker';
 import {GitHub} from '@docker/actions-toolkit/lib/github';
 import {Toolkit} from '@docker/actions-toolkit/lib/toolkit';
 
-import {BuilderInfo} from '@docker/actions-toolkit/lib/types/builder';
+import {BuilderInfo} from '@docker/actions-toolkit/lib/types/buildx/builder';
 import {GitHubRepo} from '@docker/actions-toolkit/lib/types/github';
 
 import * as context from '../src/context';
diff --git a/src/main.ts b/src/main.ts
index 04d5c3d..3e54589 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -10,7 +10,7 @@ import {Exec} from '@docker/actions-toolkit/lib/exec';
 import {GitHub} from '@docker/actions-toolkit/lib/github';
 import {Toolkit} from '@docker/actions-toolkit/lib/toolkit';
 
-import {ConfigFile} from '@docker/actions-toolkit/lib/types/docker';
+import {ConfigFile} from '@docker/actions-toolkit/lib/types/docker/docker';
 
 import * as context from './context';