pythonthings
diff --git a/Collapse file
‎bpython/curtsiesfrontend/repl.py‎
Copy file name to clipboardExpand all lines: bpython/curtsiesfrontend/repl.py
+7Lines changed: 7 additions & 0 deletions b/Collapse file
‎bpython/curtsiesfrontend/repl.py‎
Copy file name to clipboardExpand all lines: bpython/curtsiesfrontend/repl.py
+7Lines changed: 7 additions & 0 deletions
diff --git a/Collapse file
‎bpython/repl.py‎
Copy file name to clipboardExpand all lines: bpython/repl.py
+4-2Lines changed: 4 additions & 2 deletions b/Collapse file
‎bpython/repl.py‎
Copy file name to clipboardExpand all lines: bpython/repl.py
+4-2Lines changed: 4 additions & 2 deletions
@@ -1193,6 +1193,13 @@ def reprint_line(self, lineno, tokens):
         logger.debug("calling reprint line with %r %r", lineno, tokens)
         if self.config.syntax:
             self.display_buffer[lineno] = bpythonparse(format(tokens, self.formatter))
+
+    def take_back_buffer_line(self):
+        self.display_buffer.pop()
+        self.buffer.pop()
+        self.cursor_offset = 0
+        self.current_line = ''
+
     def reevaluate(self, insert_into_history=False):
         """bpython.Repl.undo calls this"""
         if self.watcher: self.watcher.reset()
 
@@ -902,8 +902,10 @@ def undo(self, n=1):
         entries = list(self.rl_history.entries)
 
         self.history = self.history[:-n]
-
-        self.reevaluate()
+        if n == 1 and not self.done:
+            self.take_back_buffer_line()
+        else:
+            self.reevaluate()
 
         self.rl_history.entries = entries