changed the mocking so it works with the lazy load