syntax highlighting bug fixes

1 files changed, 3 insertions, 1 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 47dbdf5..3f3daf2 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -222,9 +222,11 @@ impl Session {
 
     fn reprint_prompt(this: Arc<Mutex<Self>>) {
         io::stdout().write_all(&Self::prompt(this.clone())).unwrap();
-        let this = this.lock().unwrap();
+        let mut this = this.lock().unwrap();
         this.line.display_pre();
         this.line.display_post(b"");
+        this.line.mark_dirty();
+        this.cohere().unwrap();
     }
 
     fn display_historic_entry(&mut self) {