simpler & better mocking