field = (struct kprobe_trace_entry_head *)iter->ent;
        tp = container_of(event, struct trace_probe, call.event);
 
-       if (!trace_seq_printf(s, "%s: (", ftrace_event_name(&tp->call)))
-               goto partial;
+       trace_seq_printf(s, "%s: (", ftrace_event_name(&tp->call));
 
        if (!seq_print_ip_sym(s, field->ip, flags | TRACE_ITER_SYM_OFFSET))
-               goto partial;
+               goto out;
 
-       if (!trace_seq_putc(s, ')'))
-               goto partial;
+       trace_seq_putc(s, ')');
 
        data = (u8 *)&field[1];
        for (i = 0; i < tp->nr_args; i++)
                if (!tp->args[i].type->print(s, tp->args[i].name,
                                             data + tp->args[i].offset, field))
-                       goto partial;
-
-       if (!trace_seq_putc(s, '\n'))
-               goto partial;
+                       goto out;
 
-       return TRACE_TYPE_HANDLED;
-partial:
-       return TRACE_TYPE_PARTIAL_LINE;
+       trace_seq_putc(s, '\n');
+ out:
+       return trace_handle_return(s);
 }
 
 static enum print_line_t
        field = (struct kretprobe_trace_entry_head *)iter->ent;
        tp = container_of(event, struct trace_probe, call.event);
 
-       if (!trace_seq_printf(s, "%s: (", ftrace_event_name(&tp->call)))
-               goto partial;
+       trace_seq_printf(s, "%s: (", ftrace_event_name(&tp->call));
 
        if (!seq_print_ip_sym(s, field->ret_ip, flags | TRACE_ITER_SYM_OFFSET))
-               goto partial;
+               goto out;
 
-       if (!trace_seq_puts(s, " <- "))
-               goto partial;
+       trace_seq_puts(s, " <- ");
 
        if (!seq_print_ip_sym(s, field->func, flags & ~TRACE_ITER_SYM_OFFSET))
-               goto partial;
+               goto out;
 
-       if (!trace_seq_putc(s, ')'))
-               goto partial;
+       trace_seq_putc(s, ')');
 
        data = (u8 *)&field[1];
        for (i = 0; i < tp->nr_args; i++)
                if (!tp->args[i].type->print(s, tp->args[i].name,
                                             data + tp->args[i].offset, field))
-                       goto partial;
+                       goto out;
 
-       if (!trace_seq_putc(s, '\n'))
-               goto partial;
+       trace_seq_putc(s, '\n');
 
-       return TRACE_TYPE_HANDLED;
-partial:
-       return TRACE_TYPE_PARTIAL_LINE;
+ out:
+       return trace_handle_return(s);
 }