Update for test coverage