pr feedback

remix-run · Feb 21, 2024 · 9b395f9 · 9b395f9
1 parent a7d62c4
commit 9b395f9
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/packages/remix-dev/vite/get-server-build.ts b/packages/remix-dev/vite/get-server-build.ts
@@ -1,15 +1,17 @@
 import type { ViteDevServer } from "vite";
 import type { ServerBuild } from "@remix-run/server-runtime";
 
-const devServerBuild = "virtual:remix/server-build";
-
 export async function getServerBuild(
   path: string,
   viteDevServer?: ViteDevServer
-): Promise<ServerBuild> {
+): Promise<ServerBuild | (() => Promise<ServerBuild>)> {
   if (viteDevServer) {
-    return viteDevServer.ssrLoadModule(devServerBuild) as Promise<ServerBuild>;
+    return () =>
+      viteDevServer.ssrLoadModule(
+        "virtual:remix/server-build"
+      ) as Promise<ServerBuild>;
   }
+
   try {
     return import(path);
   } catch (error) {