improving test coverage pt2