Minor fix for compilation