Update test and add test coverage.