Fix issue with function head mismatch