diff --git a/javascript/babel.config.cjs b/javascript/babel.config.cjs
index 70edd37d..7487d10e 100644
--- a/javascript/babel.config.cjs
+++ b/javascript/babel.config.cjs
@@ -23,7 +23,7 @@ module.exports = api => ({
     [
       '@babel/preset-typescript',
       {
-        rewriteImportExtensions: !api.env('test'),
+        rewriteImportExtensions: api.env('dist'),
       },
     ],
   ],
diff --git a/javascript/just.config.cjs b/javascript/just.config.cjs
index 634fc523..ce8b91e5 100644
--- a/javascript/just.config.cjs
+++ b/javascript/just.config.cjs
@@ -107,6 +107,10 @@ function babelTransformTask(opts) {
 
     return spawn(node, [require.resolve('@babel/cli/bin/babel'), ...args], {
       cwd: __dirname,
+      env: {
+        // Trigger distribution-specific Babel transforms
+        NODE_ENV: 'dist',
+      }
     });
   };
 }