Started using mocks in tests