How to keep mocks consistent?