When a client select different variant from the dropdown menu in the product details, the page refreshes and the selection is set back to the previous variant.
The same thing is happening in your demo too.

Please, check and let me know is it supposed to function that way or it's a bug.

Thank you