Minor fix for previous commit