Update code to get code coverage back up