Update query_tests.py

changed