struct trace_entry *ent = iter->ent;
        struct syscall_trace_enter *trace;
        struct syscall_metadata *entry;
-       int i, ret, syscall;
+       int i, syscall;
 
        trace = (typeof(trace))ent;
        syscall = trace->nr;
                goto end;
        }
 
-       ret = trace_seq_printf(s, "%s(", entry->name);
-       if (!ret)
-               return TRACE_TYPE_PARTIAL_LINE;
+       trace_seq_printf(s, "%s(", entry->name);
 
        for (i = 0; i < entry->nb_args; i++) {
+
+               if (trace_seq_has_overflowed(s))
+                       goto end;
+
                /* parameter types */
-               if (trace_flags & TRACE_ITER_VERBOSE) {
-                       ret = trace_seq_printf(s, "%s ", entry->types[i]);
-                       if (!ret)
-                               return TRACE_TYPE_PARTIAL_LINE;
-               }
+               if (trace_flags & TRACE_ITER_VERBOSE)
+                       trace_seq_printf(s, "%s ", entry->types[i]);
+
                /* parameter values */
-               ret = trace_seq_printf(s, "%s: %lx%s", entry->args[i],
-                                      trace->args[i],
-                                      i == entry->nb_args - 1 ? "" : ", ");
-               if (!ret)
-                       return TRACE_TYPE_PARTIAL_LINE;
+               trace_seq_printf(s, "%s: %lx%s", entry->args[i],
+                                trace->args[i],
+                                i == entry->nb_args - 1 ? "" : ", ");
        }
 
-       ret = trace_seq_putc(s, ')');
-       if (!ret)
-               return TRACE_TYPE_PARTIAL_LINE;
-
+       trace_seq_putc(s, ')');
 end:
-       ret =  trace_seq_putc(s, '\n');
-       if (!ret)
-               return TRACE_TYPE_PARTIAL_LINE;
+       trace_seq_putc(s, '\n');
 
-       return TRACE_TYPE_HANDLED;
+       return trace_handle_return(s);
 }
 
 static enum print_line_t
        struct syscall_trace_exit *trace;
        int syscall;
        struct syscall_metadata *entry;
-       int ret;
 
        trace = (typeof(trace))ent;
        syscall = trace->nr;
 
        if (!entry) {
                trace_seq_putc(s, '\n');
-               return TRACE_TYPE_HANDLED;
+               goto out;
        }
 
        if (entry->exit_event->event.type != ent->type) {
                return TRACE_TYPE_UNHANDLED;
        }
 
-       ret = trace_seq_printf(s, "%s -> 0x%lx\n", entry->name,
+       trace_seq_printf(s, "%s -> 0x%lx\n", entry->name,
                                trace->ret);
-       if (!ret)
-               return TRACE_TYPE_PARTIAL_LINE;
 
-       return TRACE_TYPE_HANDLED;
+ out:
+       return trace_handle_return(s);
 }
 
 extern char *__bad_type_size(void);