Adds enough mocking to avoid null pointer