Minor update to address logging issue