Minor fix in query.py