diff --git a/jest.config.js b/jest.config.js
index 27f644ab..104b2dc1 100644
--- a/jest.config.js
+++ b/jest.config.js
@@ -21,10 +21,10 @@ switch (process.env.npm_lifecycle_event) {
   default:
     process.env.TRACE_LEVEL = 'log'
     moduleNameMapper = {
-      '^\\./dist$': '<rootDir>/src/index.js',
-      '^\\./dist/types(\\.js)?$': '<rootDir>/src/types.js',
+      '^\\./dist$': '<rootDir>/src/index.ts',
+      '^\\./dist/types(\\.js)?$': '<rootDir>/src/types.ts',
       '^\\./dist/(.+)$': '<rootDir>/src/$1',
-      '^\\.\\./dist/test-events.js$': '<rootDir>/src/test-events.js'
+      '^\\.\\./dist/test-events.js$': '<rootDir>/src/test-events.ts'
     }
 }
 
diff --git a/tests/is-node.ts b/tests/is-node.ts
index fda3e5d2..4307d9e6 100644
--- a/tests/is-node.ts
+++ b/tests/is-node.ts
@@ -1,4 +1,4 @@
-import { Alias, Pair, Scalar, YAMLMap, YAMLSeq } from '../dist/types.js'
+import { Alias, Pair, Scalar, YAMLMap, YAMLSeq } from '../types.js'
 import {
   Document,
   isAlias,