add query in tests