Swap Token not switching icons

Hello - thank you for making this awesome ui framework. I have noticed an issue with the swap example. The example here works well when the switch button is pressed: https://cosmology.zone/components?id=swap-token .

but the when building the example from this repo, the icons do not switch. however the balance does switch over. I tried several things to get it to work, but was not successful.