MainLoop.start/stop now start/stop the screen, too.

1 files changed, 5 insertions, 2 deletions

diff --git a/urwid/main_loop.py b/urwid/main_loop.py
index 967596d..613f5c4 100755
--- a/urwid/main_loop.py
+++ b/urwid/main_loop.py
@@ -275,8 +275,7 @@ class MainLoop(object):
         and :meth:`stop` once it's finished.
         """
         try:
-            with self.screen.start():
-                self._run()
+            self._run()
         except ExitMainLoop:
             pass
 
@@ -327,6 +326,8 @@ class MainLoop(object):
         Twisted and asyncio loops won't stop automatically when
         :exc:`ExitMainLoop` (or anything else) is raised.
         """
+        self.screen.start()
+
         if self.handle_mouse:
             self.screen.set_mouse_tracking()
 
@@ -356,6 +357,8 @@ class MainLoop(object):
             self._reset_input_descriptors)
         self.screen.unhook_event_loop(self.event_loop)
 
+        self.screen.stop()
+
     def _reset_input_descriptors(self):
         self.screen.unhook_event_loop(self.event_loop)
         self.screen.hook_event_loop(self.event_loop, self._update)