Correctly mocking results now