1 files changed, 5 insertions, 3 deletions
diff --git a/src/server/Initialization.ts b/src/server/Initialization.ts
index ff2b64317..b58bc3e70 100644
--- a/src/server/Initialization.ts
+++ b/src/server/Initialization.ts
@@ -18,8 +18,8 @@ import * as whm from 'webpack-hot-middleware';
 import * as fs from 'fs';
 import * as request from 'request';
 import RouteSubscriber from './RouteSubscriber';
-import { publicDirectory } from '.';
-import { logPort, addBeforeExitHandler } from './ActionUtilities';
+import { publicDirectory, ExitHandlers } from '.';
+import { logPort, } from './ActionUtilities';
 import { timeMap } from './ApiManagers/UserManager';
 import { blue, yellow } from 'colors';
 
@@ -31,6 +31,8 @@ export interface InitializationOptions {
     routeSetter: RouteSetter;
 }
 
+export let disconnect: Function;
+
 export default async function InitializeServer(options: InitializationOptions) {
     const { serverPort, routeSetter } = options;
     const app = buildWithMiddleware(express());
@@ -65,7 +67,7 @@ export default async function InitializeServer(options: InitializationOptions) {
         logPort("server", serverPort);
         console.log();
     });
-    addBeforeExitHandler(async () => { await new Promise<Error>(resolve => server.close(resolve)); });
+    disconnect = async () => new Promise<Error>(resolve => server.close(resolve));
 
     return isRelease;
 }