Minor fix for last commit