diff --git a/package.json b/package.json
index a55b8f8..1622c33 100644
--- a/package.json
+++ b/package.json
@@ -23,21 +23,34 @@
   ],
   "exports": {
     ".": {
+      "vdev": "./src/index.ts",
       "types": "./dist/index.d.ts",
       "default": "./dist/index.js"
     },
     "./pure": {
+      "vdev": "./src/pure.tsx",
       "types": "./dist/pure.d.ts",
       "default": "./dist/pure.js"
     },
     "./package.json": "./package.json"
   },
+  "publishConfig": {
+    "exports": {
+      ".": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "./pure": {
+        "types": "./dist/pure.d.ts",
+        "default": "./dist/pure.js"
+      },
+      "./package.json": "./package.json"
+    }
+  },
   "main": "./dist/index.js",
   "module": "./dist/index.js",
   "types": "./dist/index.d.ts",
   "files": [
-    "*.d.ts",
-    "*.mjs",
     "dist"
   ],
   "scripts": {
diff --git a/test/render-hook.test.tsx b/test/render-hook.test.tsx
index 1c06634..55ec50a 100644
--- a/test/render-hook.test.tsx
+++ b/test/render-hook.test.tsx
@@ -1,7 +1,7 @@
 import { expect, test } from 'vitest'
 import type { PropsWithChildren } from 'react'
 import React from 'react'
-import { renderHook } from '../src/index'
+import { renderHook } from 'vitest-browser-react'
 import { useCounter } from './fixtures/useCounter'
 
 test('should increment counter', async () => {
diff --git a/test/render.test.tsx b/test/render.test.tsx
index f0cd043..2db82a6 100644
--- a/test/render.test.tsx
+++ b/test/render.test.tsx
@@ -2,7 +2,7 @@ import { expect, test, vi } from 'vitest'
 import { page, userEvent } from 'vitest/browser'
 import { Button } from 'react-aria-components'
 import { Suspense } from 'react'
-import { render } from '../src/index'
+import { render } from 'vitest-browser-react'
 import { HelloWorld } from './fixtures/HelloWorld'
 import { Counter } from './fixtures/Counter'
 import { SuspendedHelloWorld } from './fixtures/SuspendedHelloWorld'
diff --git a/vitest.config.ts b/vitest.config.ts
index 48c264e..38abbed 100644
--- a/vitest.config.ts
+++ b/vitest.config.ts
@@ -4,7 +4,19 @@ import { playwright } from '@vitest/browser-playwright'
 
 export default defineConfig({
   plugins: [react()],
+  optimizeDeps: { include: ['vitest-browser-react'] },
   test: {
+    projects: [
+      {
+        extends: true,
+        test: { name: 'prod' },
+      },
+      {
+        extends: true,
+        test: { name: 'dev' },
+        resolve: { conditions: ['vdev'] },
+      },
+    ],
     printConsoleTrace: true,
     browser: {
       enabled: true,