Make test passing with 100% coverage