Fix task2 query.py