diff --git a/tests/unit/ant.test.ts b/tests/unit/ant.test.ts
index af10391c..af666aa5 100644
--- a/tests/unit/ant.test.ts
+++ b/tests/unit/ant.test.ts
@@ -1,7 +1,7 @@
 import { strict as assert } from 'node:assert';
 import { describe, it } from 'node:test';
 
-import { isAoANTState } from '../../src/types/ant';
+import { isAoANTState } from '../../src/types/ant.js';
 
 const testAoANTState = {
   Name: 'TestToken',
diff --git a/tests/unit/token.test.ts b/tests/unit/token.test.ts
index 36c9b401..b8dc277a 100644
--- a/tests/unit/token.test.ts
+++ b/tests/unit/token.test.ts
@@ -1,7 +1,7 @@
 import { strict as assert } from 'node:assert';
 import { describe, it } from 'node:test';
 
-import { IOToken, mIOToken } from '../../src/types/token';
+import { IOToken, mIOToken } from '../../src/types/token.js';
 
 describe('IOToken', () => {
   it('should throw an error on invalid input', () => {