update mocks so tests will pass