Fix highlights for queries.

Signed-off-by: Moritz Ulrich <moritz@tarn-vedra.de>