Minor fix on includes