diff --git a/package.json b/package.json
index 852938f..d8c2538 100644
--- a/package.json
+++ b/package.json
@@ -1,8 +1,9 @@
 {
     "name": "remark-torchlight",
-    "version": "0.0.2",
+    "version": "0.0.3",
     "description": "A remark plugin for Torchlight - the syntax highlighting API.",
-    "main": "index.js",
+    "exports": "./index.js",
+	"type": "module",
     "scripts": {
         "test": "standard --env jest && jest"
     },
diff --git a/tests/__snapshots__/render.test.js.snap b/tests/__snapshots__/render.test.js.snap
index 96fe56e..a30e33f 100644
--- a/tests/__snapshots__/render.test.js.snap
+++ b/tests/__snapshots__/render.test.js.snap
@@ -2,6 +2,6 @@
 
 exports[`it tests 1`] = `
 "<p>this <em>is</em> <a href=\\"https://www.google.com\\">test</a></p>
-<pre class=\\"classes\\" style=\\"style: 1;\\"><code class=\\"language-js\\">highlighted</code></pre>
+<pre><code>highlighted</code></pre>
 "
 `;
diff --git a/tests/render.test.js b/tests/render.test.js
index 50c6ba4..3b983f7 100644
--- a/tests/render.test.js
+++ b/tests/render.test.js
@@ -1,4 +1,4 @@
-import torchlight from '../index'
+import torchlight from '../index.js'
 import { remark } from 'remark'
 import html from 'remark-html'
 import {mockApi} from '@torchlight-api/torchlight-cli/tests/support/helpers'