fixed issue with passing of mock to retriever