diff --git a/code/lib/builder-vite/src/index.ts b/code/lib/builder-vite/src/index.ts
index 8976a534072..b9535c4a5e0 100644
--- a/code/lib/builder-vite/src/index.ts
+++ b/code/lib/builder-vite/src/index.ts
@@ -69,14 +69,8 @@ function iframeMiddleware(options: ExtendedOptions, server: ViteDevServer): Requ
 
 let server: ViteDevServer;
 
-export async function bail(e?: Error): Promise<void> {
-  try {
-    return await server.close();
-  } catch (err) {
-    console.warn('unable to close vite server');
-  }
-
-  throw e;
+export async function bail(): Promise<void> {
+  return server?.close();
 }
 
 export const start: ViteBuilder['start'] = async ({
diff --git a/docs/snippets/common/storybook-builder-api-shutdown-server.ts.mdx b/docs/snippets/common/storybook-builder-api-shutdown-server.ts.mdx
index 4c92faa541f..429f8eff4e5 100644
--- a/docs/snippets/common/storybook-builder-api-shutdown-server.ts.mdx
+++ b/docs/snippets/common/storybook-builder-api-shutdown-server.ts.mdx
@@ -3,19 +3,16 @@
 
 import { createViteServer } from './vite-server';
 
+
+let server: ViteDevServer;
+export async function bail(): Promise<void> {
+  return server?.close();
+}
+
 export const start: ViteBuilder['start'] = async ({ options, server: devServer }) => {
   // Remainder implementation goes here
-
-  const server = await createViteServer(options as ExtendedOptions, devServer);
-  async function bail(e?: Error): Promise<void> {
-    try {
-      return await server.close();
-    } catch (err) {
-      console.warn('unable to close the server');
-    }
-    throw e;
-  }
-
+  server = await createViteServer(options as ExtendedOptions, devServer);
+  
   return {
     bail,
     totalTime: process.hrtime(startTime),