Fix test coverage