I wonder if we should just ship all headers from `ui/`? I guess we'll see if we need to add any more in the future.