Update test coverage