Minor fix to logging

Minor fix to logging