I think a little improvement can be here, on the screenshot, when the screen size gets narrower (less than 400px) the layout becomes inconsistent.
I think this can be a better way to do it.
If you think too I already made a fix locally, I will just create a PR so it can be merged after approval.