diff --git a/addons/viewport/package.json b/addons/viewport/package.json
index 9c3816d33e5e..ca54026ceb43 100644
--- a/addons/viewport/package.json
+++ b/addons/viewport/package.json
@@ -23,7 +23,8 @@
     "*.js",
     "*.d.ts"
   ],
-  "main": "preview.js",
+  "main": "dist/preview.js",
+  "types": "dist/preview.d.ts",
   "scripts": {
     "prepare": "node ../../scripts/prepare.js"
   },
diff --git a/addons/viewport/preview.js b/addons/viewport/preview.js
deleted file mode 100644
index dc47e9c63cba..000000000000
--- a/addons/viewport/preview.js
+++ /dev/null
@@ -1,5 +0,0 @@
-const preview = require('./dist/legacy_preview/index');
-
-exports.configureViewport = preview.configureViewport;
-exports.DEFAULT_VIEWPORT = preview.DEFAULT_VIEWPORT;
-exports.INITIAL_VIEWPORTS = preview.INITIAL_VIEWPORTS;
diff --git a/addons/viewport/src/preview.ts b/addons/viewport/src/preview.ts
new file mode 100644
index 000000000000..be2dc03d34c4
--- /dev/null
+++ b/addons/viewport/src/preview.ts
@@ -0,0 +1 @@
+export { configureViewport, DEFAULT_VIEWPORT, INITIAL_VIEWPORTS } from './legacy_preview';