fix test mocks