tracing: Have branch tracer use trace_handle_return() helper function