trying to fix issue with test coverage