diff --git a/__tests__/buildx/buildx.test.ts b/__tests__/buildx/buildx.test.ts
index e11f62d..a876780 100644
--- a/__tests__/buildx/buildx.test.ts
+++ b/__tests__/buildx/buildx.test.ts
@@ -303,7 +303,7 @@ describe('localState', () => {
       } as LocalState,
     ]
   ])('given %p', async (ref: string, expected: LocalState) => {
-    const localState = Buildx.localState(path.join(fixturesDir, 'buildx-refs'), ref);
+    const localState = Buildx.localState(ref, path.join(fixturesDir, 'buildx-refs'));
     expect(localState).toEqual(expected);
   });
 });
diff --git a/src/buildx/buildx.ts b/src/buildx/buildx.ts
index 728d9dd..97eb4f8 100644
--- a/src/buildx/buildx.ts
+++ b/src/buildx/buildx.ts
@@ -177,12 +177,12 @@ export class Buildx {
     return driverOpts;
   }
 
-  public static localState(dir: string, ref: string): LocalState {
+  public static localState(ref: string, dir?: string): LocalState {
     const [builderName, nodeName, id] = ref.split('/');
     if (!builderName || !nodeName || !id) {
       throw new Error(`Invalid build reference: ${ref}`);
     }
-    const lsPath = path.join(dir, builderName, nodeName, id);
+    const lsPath = path.join(dir || Buildx.refsDir, builderName, nodeName, id);
     if (!fs.existsSync(lsPath)) {
       throw new Error(`Local state not found in ${lsPath}`);
     }