diff --git a/src/visual-regression-app/vite.config.ts b/src/visual-regression-app/vite.config.ts
index d6fed16a09..98aea914bb 100644
--- a/src/visual-regression-app/vite.config.ts
+++ b/src/visual-regression-app/vite.config.ts
@@ -186,7 +186,9 @@ function prepareScreenshots(): PluginOption {
                   };
                 });
             });
-        } else if (viteConfig.command !== 'serve') {
+        }
+
+        if (viteConfig.command !== 'serve') {
           this.emitFile({
             type: 'asset',
             fileName: 'diff.json',