diff --git a/package.json b/package.json
index f0115f0..046a963 100644
--- a/package.json
+++ b/package.json
@@ -10,7 +10,7 @@
   "types": "lib/index.d.ts",
   "exports": {
     ".": "./lib/index.js",
-    "./*": "./"
+    "./*": "./*"
   },
   "repository": "prettier/yaml-unist-parser",
   "homepage": "https://github.com/prettier/yaml-unist-parser#readme",
diff --git a/src/yaml.ts b/src/yaml.ts
index a57d22c..38b2f8d 100644
--- a/src/yaml.ts
+++ b/src/yaml.ts
@@ -3,18 +3,22 @@
 // tslint:disable:variable-name
 
 import YAML from "yaml";
-import { CST } from "yaml/parse-cst.js";
+// eslint-disable-next-line import/extensions
+import { CST } from "yaml/parse-cst";
 import {
   Alias as _Alias,
   Merge as _Merge,
   Node as _Node,
   Pair as _Pair,
   Scalar as _Scalar,
-} from "yaml/types.js";
+  // eslint-disable-next-line import/extensions
+} from "yaml/types";
 
 export const { Document, parseCST } = YAML;
-export { ParsedCST } from "yaml/parse-cst.js";
-export { YAMLError, YAMLSyntaxError, YAMLSemanticError } from "yaml/util.js";
+// eslint-disable-next-line import/extensions
+export { ParsedCST } from "yaml/parse-cst";
+// eslint-disable-next-line import/extensions
+export { YAMLError, YAMLSyntaxError, YAMLSemanticError } from "yaml/util";
 
 export namespace ast {
   export type Alias = _Alias;