fix some test coverage issues