I think I fixed the test coverage problems