Return buffer before sending it to the server

1 files changed, 2 insertions, 1 deletions

diff --git a/pgcli/main.py b/pgcli/main.py
index d86c247a..7ec2606c 100755
--- a/pgcli/main.py
+++ b/pgcli/main.py
@@ -183,7 +183,8 @@ class PGCli(object):
             if message:
                 # Something went wrong. Raise an exception and bail.
                 raise RuntimeError(message)
-            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