improve test coverage on main