Adding more coverage to tests.