I’m trying a very simple animation to swap two Divs position with @react-spring/web but I can’t. It seems partially working but I don’t understand why.

I am working on a small project and I am trying to implement shaking animation when the user clicks dont know. Right now, when I click the button it shakes two times. It does that when i add onRest function. When I remove it, it works great but I need to use that button multiple times so i need to update the state after the animation executes. How to solve this?