Adding mocking for the init