Could you please add test coverage for this methods?