Make arrow_box not overwrite className in traces