fixing test mocking