Prettified the mocks a little more