updated mocks in tests