Minor fix to the previous commit