1 files changed, 2 insertions, 2 deletions

diff --git a/samples/bpf/tracex2_kern.c b/samples/bpf/tracex2_kern.c
index 09c1adc27d42..6d6eefd0d465 100644
--- a/samples/bpf/tracex2_kern.c
+++ b/samples/bpf/tracex2_kern.c
@@ -27,10 +27,10 @@ int bpf_prog2(struct pt_regs *ctx)
         long init_val = 1;
         long *value;
 
-        /* x64/s390x specific: read ip of kfree_skb caller.
+        /* read ip of kfree_skb caller.
          * non-portable version of __builtin_return_address(0)
          */
-        bpf_probe_read(&loc, sizeof(loc), (void *)PT_REGS_RET(ctx));
+        BPF_KPROBE_READ_RET_IP(loc, ctx);
 
         value = bpf_map_lookup_elem(&my_map, &loc);
         if (value)