Changed use of mock to double