Assistant doesn't remember last selected model in some situations

Reproduction steps

Ask any query via Chat
In the results page, try switching between Research and Chat
Observe that Research and Chat's model is set to its default value, despite it being set (and remembered) correctly on the home page.

I expect Kagi Assistant to also use the last used model when switching between modes after asking the initial query.