diff --git a/packages/compiler/src/setup-transformer.ts b/packages/compiler/src/setup-transformer.ts
index f1a1078..748eb6f 100644
--- a/packages/compiler/src/setup-transformer.ts
+++ b/packages/compiler/src/setup-transformer.ts
@@ -1,10 +1,6 @@
-import { fileURLToPath, pathToFileURL } from "node:url";
 import ts from "typescript";
 
 const tempblotInstanceVarName = "__tempblotInstance";
-const tempblotModulePath = pathToFileURL(
-  fileURLToPath(new URL("./index.ts", import.meta.url)),
-).href;
 
 function isTempblotImportDeclaration(
   node: ts.Node,
@@ -93,7 +89,7 @@ function updateTempblotImportDeclaration(
       node,
       node.modifiers,
       importClause,
-      ts.factory.createStringLiteral(tempblotModulePath),
+      node.moduleSpecifier,
       node.attributes,
     );
   }
@@ -109,7 +105,7 @@ function updateTempblotImportDeclaration(
       node,
       node.modifiers,
       importClause,
-      ts.factory.createStringLiteral(tempblotModulePath),
+      node.moduleSpecifier,
       node.attributes,
     );
   }
@@ -132,7 +128,7 @@ function updateTempblotImportDeclaration(
     node,
     node.modifiers,
     updatedImportClause,
-    ts.factory.createStringLiteral(tempblotModulePath),
+    node.moduleSpecifier,
     node.attributes,
   );
 }
diff --git a/packages/compiler/tests/__snapshots__/setup-transformer.spec.ts.snap b/packages/compiler/tests/__snapshots__/setup-transformer.spec.ts.snap
index 9619c32..42d0811 100644
--- a/packages/compiler/tests/__snapshots__/setup-transformer.spec.ts.snap
+++ b/packages/compiler/tests/__snapshots__/setup-transformer.spec.ts.snap
@@ -6,7 +6,7 @@ exports[`does not transform setup without tempblot useParams import 1`] = `
 `;
 
 exports[`transforms aliased useParams calls 1`] = `
-"import { useParams as getParams, TempblotInstance } from "file:///tempblot/index.ts";
+"import { useParams as getParams, TempblotInstance } from "tempblot";
 const __tempblotInstance = new TempblotInstance(globalThis.tempblotParams["/path/to/somefile.tempblot"]);
 const config = getParams.call(__tempblotInstance) as {
     abc: 1;
@@ -15,7 +15,7 @@ const config = getParams.call(__tempblotInstance) as {
 `;
 
 exports[`transforms useParams calls 1`] = `
-"import { useParams, TempblotInstance } from "file:///tempblot/index.ts";
+"import { useParams, TempblotInstance } from "tempblot";
 const __tempblotInstance = new TempblotInstance(globalThis.tempblotParams["/path/to/somefile.tempblot"]);
 const _config = useParams.call(__tempblotInstance);
 const config = useParams.call(__tempblotInstance) as {
diff --git a/packages/compiler/tests/setup-transformer.spec.ts b/packages/compiler/tests/setup-transformer.spec.ts
index f7fb441..6bc0540 100644
--- a/packages/compiler/tests/setup-transformer.spec.ts
+++ b/packages/compiler/tests/setup-transformer.spec.ts
@@ -2,11 +2,6 @@ import { expect, test } from "vitest";
 import { transformSetup } from "../src/setup-transformer.js";
 
 const sourcePath = "/path/to/somefile.tempblot";
-const tempblotModulePath = new URL("../src/index.ts", import.meta.url).href;
-
-function normalizeOutput(output: string): string {
-  return output.replaceAll(tempblotModulePath, "file:///tempblot/index.ts");
-}
 
 test("transforms useParams calls", () => {
   const result = transformSetup(
@@ -18,7 +13,7 @@ const config = useParams<{ abc: 1 }>();
     sourcePath,
   );
 
-  expect(normalizeOutput(result)).toMatchSnapshot();
+  expect(result).toMatchSnapshot();
 });
 
 test("transforms aliased useParams calls", () => {
@@ -30,7 +25,7 @@ const config = getParams<{ abc: 1 }>();
     sourcePath,
   );
 
-  expect(normalizeOutput(result)).toMatchSnapshot();
+  expect(result).toMatchSnapshot();
 });
 
 test("does not transform setup without tempblot useParams import", () => {
diff --git a/packages/compiler/vite.config.ts b/packages/compiler/vite.config.ts
index ee06098..d547839 100644
--- a/packages/compiler/vite.config.ts
+++ b/packages/compiler/vite.config.ts
@@ -10,6 +10,11 @@ export default defineConfig({
     tsconfigPath: "tsconfig.app.json",
     entryRoot: "src",
   })],
+  resolve: {
+    alias: {
+      tempblot: resolve(__dirname, "src/index.ts"),
+    },
+  },
   build: {
     lib: {
       name: "Tempblot",