Multi-device input in UI Toolkit

Hi,
My team and I decided to use the new UI Toolkit for a 2-player couch co-op game we’re developing.
We’d like to allow both players to control the main menu - as in, navigating on either controller moves the focus around. However, only the first player that connects can control.
Is there some way to allow this? We’re using the new input system package and a standard event system.
Thanks in advance for any help on this!