Minor fix for build