From bcc76ff21a7e7f0ca8f97b77d41d077505ee80cd Mon Sep 17 00:00:00 2001
From: noah morrison <noah@morrison.ph>
Date: Mon, 23 Mar 2015 22:39:20 -0400
Subject: [PATCH] Some minor code cleanup.

---
 rtv/submission.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/rtv/submission.py b/rtv/submission.py
index af883a6..20229b8 100644
--- a/rtv/submission.py
+++ b/rtv/submission.py
@@ -247,14 +247,12 @@ class SubmissionPage(BasePage):
             curses.flash()
             return
 
-        curses.endwin()
 
         cid = str(uuid4())
         if data['type'] == 'Submission':
             info = (data['author'], 'submission', data['text'])
         else:
             info = (data['author'], 'comment', data['body'])
-
         comment_info = COMMENT_FILE.format(cid, *info)
 
         with NamedTemporaryFile(prefix='rtv-comment-', mode='w+') as fp:
@@ -266,12 +264,13 @@ class SubmissionPage(BasePage):
                 show_notification(self.stdscr, ['No EDITOR defined'])
                 return
 
+            curses.endwin()
             os.system(editor + ' ' + fp.name)
+            curses.doupdate()
 
             fp.seek(0)
             comment_text = fp.read().split('--% ' + cid + ' %--')[0]
 
-        curses.doupdate()
 
         if comment_text is None or comment_text.isspace():
             return