Minor fix to project