Minor fix for the previous commit