Fix function/class case