Same deal on my MacBook Air M1. Restarting the mouse works but it's super annoying so I hope they fix it in the next update.