diff --git a/Python/btui.py b/Python/btui.py
index ce923cd..9b39bc0 100644
--- a/Python/btui.py
+++ b/Python/btui.py
@@ -196,9 +196,10 @@ class BTUI:
     @contextmanager
     def buffered(self):
         assert self._btui
+        prev_autoflush = self._autoflush
         self._autoflush = False
         yield
-        self._autoflush = True
+        self._autoflush = prev_autoflush
         self.flush()
 
     def getkey(self, timeout=None):