Returned to 100% test coverage