Add middle of the line completion.

1 files changed, 3 insertions, 3 deletions

diff --git a/pgcli/pgcompleter.py b/pgcli/pgcompleter.py
index 44d11267..e50edf8c 100644
--- a/pgcli/pgcompleter.py
+++ b/pgcli/pgcompleter.py
@@ -78,11 +78,11 @@ class PGCompleter(Completer):
         # will always be the partially typed string which renders the smart
         # completion useless because it will always return the list of keywords
         # as completion.
-
         if word_before_cursor:
-            parsed = sqlparse.parse(document.text[:-len(word_before_cursor)])
+            parsed = sqlparse.parse(
+                    document.text_before_cursor[:-len(word_before_cursor)])
         else:
-            parsed = sqlparse.parse(document.text)
+            parsed = sqlparse.parse(document.text_before_cursor)
 
         last_token = ''
         if parsed: