retrieval: improve test coverage