Minor fix for compile