Partial revert

src/asyncclick/core.py

@@ -624,7 +624,7 @@ def call_on_close(self, f: t.Callable[..., t.Any]) -> t.Callable[..., t.Any]:
 
         :param f: The function to execute on teardown.
         """
-        return self._ctx_mgr.callback(f)
+        return self._exit_stack.callback(f)
 
     def close(self) -> None:
         """Invoke all close callbacks registered with