diff --git a/test/integration/link.js b/test/integration/link.js
index 680e604b..a5f2b95a 100644
--- a/test/integration/link.js
+++ b/test/integration/link.js
@@ -12,8 +12,12 @@ class TestControl extends link.Link {
 		link.attachAllMessages(this);
 	}
 
+	async prepareDisconnectRequestHandler(message, request) {
+		this.connector.setClosing();
+		return await super.prepareDisconnectRequestHandler(message, request);
+	}
+
 	async debugWsMessageEventHandler() { }
-	async prepareDisconnectRequestHandler() { }
 	async instanceOutputEventHandler() { }
 }
 
@@ -64,8 +68,4 @@ describe("Integration of lib/link", function() {
 		await events.once(controlConnector, "invalidate");
 		await events.once(controlConnector, "connect");
 	});
-
-	after(async function() {
-		controlConnector.close(1001, "Test Quit");
-	});
 });