Fix the tests properly. We use our own Mocks.