Change test case for new method of query.