Minor fix to previous commit