increase test coverage on models