Minor fix, add log