Remember search term after viewing album/playlist and pressing the back button

When searching my playlists and favourited albums I often like to open one of the search results, add a few song to a playlist and then return back to the album or playlist page.

Whenever I do this the search term is forgotten and I have to type it back in. This is inconvenient and I imagine fairly easy to implement a fix for. I haven’t tested this but I believe this is universal for all search contexts expect the main search bar.
 

This happens on both MacOS and IOS for me.