diff --git a/src/server.ts b/src/server.ts index 7c118f0..ce476b1 100644 --- a/src/server.ts +++ b/src/server.ts @@ -82,8 +82,10 @@ export async function createServerWithTools(options: Options): Promise { return { contents }; }); + // Store reference to the original close method before overriding + const originalClose = server.close.bind(server); server.close = async () => { - await server.close(); + await originalClose(); await wss.close(); await context.close(); };