diff --git a/tfjs-backend-wasm/scripts/create-worker-module.js b/tfjs-backend-wasm/scripts/create-worker-module.js
index f8b3f8730fa..6de869e1b8c 100755
--- a/tfjs-backend-wasm/scripts/create-worker-module.js
+++ b/tfjs-backend-wasm/scripts/create-worker-module.js
@@ -16,9 +16,8 @@
  */
 
 /**
- * This file creates a TypeScript module that exports the contents of the
- * Emscripten-generated WASM worker script so that it can be inlined by the
- * tf-backend-wasm bundle.
+ * This file modifies the Emscripten-generated WASM worker script so that it can
+ * be inlined by the tf-backend-wasm bundle.
  */
 
 const fs = require('fs');
@@ -26,10 +25,6 @@ const fs = require('fs');
 const BASE_PATH = './wasm-out/';
 const WORKER_PATH = `${BASE_PATH}tfjs-backend-wasm-threaded-simd.worker.js`;
 
-// Write out a worker TypeScript module.
 const workerContents = fs.readFileSync(WORKER_PATH, "utf8");
-fs.writeFileSync(`${BASE_PATH}tfjs-backend-wasm-threaded-simd.worker.ts`,
+fs.writeFileSync(`${WORKER_PATH}`,
   `export const wasmWorkerContents = '${workerContents.trim()}';`);
-
-// Delete the original worker file.
-fs.unlinkSync(WORKER_PATH)
diff --git a/tfjs-backend-wasm/src/backend_wasm.ts b/tfjs-backend-wasm/src/backend_wasm.ts
index 1c8281e7046..395a6921362 100644
--- a/tfjs-backend-wasm/src/backend_wasm.ts
+++ b/tfjs-backend-wasm/src/backend_wasm.ts
@@ -21,7 +21,7 @@ import {backend_util, BackendTimingInfo, DataStorage, DataType, deprecationWarn,
 import {BackendWasmModule, WasmFactoryConfig} from '../wasm-out/tfjs-backend-wasm';
 import wasmFactoryThreadedSimd from '../wasm-out/tfjs-backend-wasm-threaded-simd.js';
 // @ts-ignore
-import {wasmWorkerContents} from '../wasm-out/tfjs-backend-wasm-threaded-simd.worker';
+import {wasmWorkerContents} from '../wasm-out/tfjs-backend-wasm-threaded-simd.worker.js';
 import wasmFactory from '../wasm-out/tfjs-backend-wasm.js';
 
 const WASM_PRIORITY = 2;