WIP tests need fixing
fix test by mocking