fixed tests to include collection in query