Update test coverage based on review feedback.