Fix build error
Minor fix