Update query.py

bug fix...