Fixed tests from failing although we now have less coverage.