Ensure Query API is loaded for all tests