Bring code coverage back up to 100%