Fix to function body code generation.