{
        int err;
        bool is_ret0 = (func == __static_call_return0);
-       unsigned long target = (unsigned long)(is_ret0 ? tramp + PPC_SCT_RET0 : func);
-       bool is_short = is_offset_in_branch_range((long)target - (long)tramp);
-
-       if (!tramp)
-               return;
+       unsigned long _tramp = (unsigned long)tramp;
+       unsigned long _func = (unsigned long)func;
+       unsigned long _ret0 = _tramp + PPC_SCT_RET0;
+       bool is_short = is_offset_in_branch_range((long)func - (long)(site ? : tramp));
 
        mutex_lock(&text_mutex);
 
-       if (func && !is_short) {
-               err = patch_ulong(tramp + PPC_SCT_DATA, target);
-               if (err)
-                       goto out;
+       if (tramp) {
+               if (func && !is_short) {
+                       err = patch_ulong(tramp + PPC_SCT_DATA, _func);
+                       if (err)
+                               goto out;
+               }
+
+               if (!func)
+                       err = patch_instruction(tramp, ppc_inst(PPC_RAW_BLR()));
+               else if (is_ret0)
+                       err = patch_branch(tramp, _ret0, 0);
+               else if (is_short)
+                       err = patch_branch(tramp, _func, 0);
+               else
+                       err = patch_instruction(tramp, ppc_inst(PPC_RAW_NOP()));
+       } else {
+               err = 0;
        }
 
-       if (!func)
-               err = patch_instruction(tramp, ppc_inst(PPC_RAW_BLR()));
-       else if (is_short)
-               err = patch_branch(tramp, target, 0);
-       else
-               err = patch_instruction(tramp, ppc_inst(PPC_RAW_NOP()));
 out:
        mutex_unlock(&text_mutex);