From 977ce7a9deeeeb13c906faf7652d7669c55c8230 Mon Sep 17 00:00:00 2001
From: alemidev <me@alemi.dev>
Date: Sat, 23 Apr 2022 15:22:04 +0200
Subject: [PATCH] reordered stop sequence

---
 src/treepuncher/treepuncher.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/treepuncher/treepuncher.py b/src/treepuncher/treepuncher.py
index 3fd5d86..1f8f7e5 100644
--- a/src/treepuncher/treepuncher.py
+++ b/src/treepuncher/treepuncher.py
@@ -243,9 +243,9 @@ class Treepuncher(
 			await self.dispatcher.disconnect(block=not force)
 		if not force:
 			await self._worker
+			await self.join_callbacks()
 			for m in self.modules:
 				await m.cleanup()
-			await self.join_callbacks()
 		await super().stop()
 		self.logger.info("Treepuncher stopped")