Minor fix to includes