fix hmr engine reference in dev (#495)

FredKSchott · Jun 15, 2020 · a047967 · a047967
1 parent 2a39016
commit a047967
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/commands/dev.ts b/src/commands/dev.ts
@@ -200,10 +200,6 @@ export async function command(commandOptions: CommandOptions) {
   const messageBus = new EventEmitter();
   const mountedDirectories: [string, string][] = [];
 
-  onProcessExit(() => {
-    hmrEngine.disconnectAllClients();
-  });
-
   console.log = (...args) => {
     messageBus.emit('CONSOLE', {level: 'log', args});
   };
@@ -871,6 +867,9 @@ export async function command(commandOptions: CommandOptions) {
     .listen(port);
 
   const hmrEngine = new EsmHmrEngine({server});
+  onProcessExit(() => {
+    hmrEngine.disconnectAllClients();
+  });
 
   // Live Reload + File System Watching
   let isLiveReloadPaused = false;