Merge pull request #241 from fpietka/fix-editmode

Fix editor mode

1 files changed, 2 insertions, 3 deletions

diff --git a/pgcli/main.py b/pgcli/main.py
index d12d5f52..7ec2606c 100755
--- a/pgcli/main.py
+++ b/pgcli/main.py
@@ -183,9 +183,8 @@ class PGCli(object):
             if message:
                 # Something went wrong. Raise an exception and bail.
                 raise RuntimeError(message)
-            document = cli.read_input(
-                initial_document=Document(sql, cursor_position=len(sql)),
-              )
+            cli.current_buffer.document = Document(sql, cursor_position=len(sql))
+            document = cli.read_input(False)
             continue
         return document