diff --git a/redash/query_runner/__init__.py b/redash/query_runner/__init__.py index 1bcb5383f3..10d6d6edf5 100644 --- a/redash/query_runner/__init__.py +++ b/redash/query_runner/__init__.py @@ -309,9 +309,11 @@ def add_limit_to_query(self, query): parsed_query.tokens += limit_tokens else: for i in range(length - 1, -1, -1): - if parsed_query[i].value.upper() == 'SELECT': + if parsed_query[i].value.upper() == "SELECT": index = parsed_query.token_index(parsed_query[i + 1]) - parsed_query = sqlparse.sql.Statement(parsed_query.tokens[:index] + limit_tokens + parsed_query.tokens[index:]) + parsed_query = sqlparse.sql.Statement( + parsed_query.tokens[:index] + limit_tokens + parsed_query.tokens[index:] + ) break return str(parsed_query)