Fix broken tests.

1 files changed, 1 insertions, 1 deletions

diff --git a/tests/utils.py b/tests/utils.py
index 91093dec..9b7376f1 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -43,7 +43,7 @@ def run(executor, sql, join=False):
     " Return string output for the sql to be run "
     result = []
     for rows, headers, status in executor.run(sql):
-        result.extend(format_output(rows, headers, status))
+        result.extend(format_output(rows, headers, status, 'psql'))
     if join:
         result = '\n'.join(result)
     return result