Add new test for query