Fix max_trace_wait docstring