Minor fix for #ifdefs