diff --git a/packages/vite/src/node/build.ts b/packages/vite/src/node/build.ts
index 086e2d2d69b585..053455b818dfc6 100644
--- a/packages/vite/src/node/build.ts
+++ b/packages/vite/src/node/build.ts
@@ -1317,7 +1317,7 @@ function wrapEnvironmentHook<HookName extends keyof Plugin>(
     if (hookName === 'augmentChunkHash') {
       injectChunkMetadata(chunkMetadataMap, args[0])
     }
-    if (hookName === 'generateBundle') {
+    if (hookName === 'generateBundle' || hookName === 'writeBundle') {
       const bundle = args[1] as OutputBundle
       for (const chunk of Object.values(bundle)) {
         if (chunk.type === 'chunk') {