* Fix test by adding mocks