Minor fix to last commit