I modify query_main.py

Change-Id: Iccdfaa0ab6dd6bbba165a94ba0c6fa6957864606