extend test coverage to more functions