fix test_query example