diff --git a/package.json b/package.json
index 1f07691..af059fe 100644
--- a/package.json
+++ b/package.json
@@ -16,21 +16,23 @@
   },
   "homepage": "https://grid.kiwiberry.nz/",
   "files": [
-    "dist",
-    "src/*.*"
+    "dist"
   ],
-  "main": "./dist/vue-virtual-scroll-grid.umd.js",
-  "module": "./dist/vue-virtual-scroll-grid.es.js",
+  "module": "./dist/index.es.js",
+  "main": "./dist/index.umd.js",
+  "types": "./dist/index.d.ts",
   "exports": {
     ".": {
-      "import": "./dist/vue-virtual-scroll-grid.es.js",
-      "require": "./dist/vue-virtual-scroll-grid.umd.js"
+      "import": "./dist/index.es.js",
+      "require": "./dist/index.umd.js",
+      "types": "./dist/index.d.ts"
     }
   },
   "license": "MIT",
   "scripts": {
+    "gen:types": "vue-tsc --declaration --emitDeclarationOnly",
     "serve": "vite preview",
-    "build": "vite build",
+    "build": "vite build && npm run gen:types",
     "build:demo": "vite build --mode demo",
     "dev": "vite",
     "lint": "vue-tsc --noEmit --skipLibCheck",
@@ -61,7 +63,6 @@
     "ts-jest": "^27.0.7",
     "typescript": "^4.5.2",
     "vite": "^2.6.13",
-    "vite-dts": "^1.0.3",
     "vue": "^3.2.23",
     "vue-tsc": ">=0.29.8"
   },
diff --git a/src/index.ts b/src/index.ts
new file mode 100644
index 0000000..77bee12
--- /dev/null
+++ b/src/index.ts
@@ -0,0 +1,3 @@
+import Grid from './Grid.vue'
+
+export default Grid
diff --git a/tsconfig.json b/tsconfig.json
index b2c0fe1..86f852e 100644
--- a/tsconfig.json
+++ b/tsconfig.json
@@ -1,6 +1,8 @@
 {
   "compilerOptions": {
-    "declaration": true,
+    "declaration": false,
+    "baseUrl": ".",
+    "outDir": "dist",
     "target": "esnext",
     "module": "esnext",
     "moduleResolution": "node",
@@ -12,5 +14,5 @@
     "lib": ["esnext", "dom"],
     "types": ["vite/client", "jest"],
   },
-  "include": ["src/*.ts", "src/*.d.ts", "src/*.tsx", "src/*.vue"]
+  "include": ["src/*.ts", "src/*.tsx", "src/*.vue"]
 }
diff --git a/vite.config.ts b/vite.config.ts
index 73fdbc8..f7253b7 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -3,11 +3,10 @@ import vue from "@vitejs/plugin-vue";
 import { homedir } from "os";
 import { existsSync, readFileSync } from "fs";
 import { resolve } from "path";
-import dts from "vite-dts";
 
 export default ({ mode }: ConfigEnv): UserConfig => {
   return {
-    plugins: [vue(), dts()],
+    plugins: [vue()],
     server: {
       open: true,
       https: existsSync(`${homedir()}/.localhost_ssl/server.key`)
@@ -23,8 +22,9 @@ export default ({ mode }: ConfigEnv): UserConfig => {
         ? {}
         : {
             lib: {
-              entry: resolve(__dirname, "src/Grid.vue"),
+              entry: resolve(__dirname, "src/index.ts"),
               name: "VirtualScrollGrid",
+              fileName: (format) => `index.${format}.js`,
             },
             rollupOptions: {
               // Make sure to externalize deps that shouldn't be bundled