Updated query in test