From 88bd5e1cf4745522b79ac8dffb3801f5c261044e Mon Sep 17 00:00:00 2001
From: Tony Garnock-Jones <tonyg@leastfixedpoint.com>
Date: Wed, 8 Sep 2021 13:07:26 +0200
Subject: [PATCH] Improved compatibility between During and Handler

---
 syndicate/during.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/syndicate/during.py b/syndicate/during.py
index 012194c..9c1ada1 100644
--- a/syndicate/during.py
+++ b/syndicate/during.py
@@ -65,6 +65,8 @@ class During(Handler):
             def facet(turn):
                 if self.inert_ok:
                     turn.prevent_inert_check()
-                handler(turn, *args)
+                maybe_stop_action = handler(turn, *args)
+                if maybe_stop_action is not None:
+                    turn.on_stop(maybe_stop_action)
             return lambda turn: turn.stop(facet)
         return facet_handler