diff --git a/src/scyjava/_script.py b/src/scyjava/_script.py index 74939f1f..ec73f906 100644 --- a/src/scyjava/_script.py +++ b/src/scyjava/_script.py @@ -47,9 +47,8 @@ def write(self, s): self._writer().write(s) def _writer(self): - return self._thread_to_context.get( - threading.currentThread(), self._std_default - ) + ctx = self._thread_to_context.get(threading.currentThread()) + return self._std_default if ctx is None else ctx.getWriter() stdoutContextWriter = ScriptContextWriter(sys.stdout)