Fix mocking/stubbing tests
2 files changed