Improve gather_messages API a bit.