I would make sure the air box lid is water tight with snorkel installed & the engine breather system is completely water tight.
I would remove the fuel tank, left frame side cover & unplug the connectors then spray every electrical connector with a water repellent spray.
I used to race enduro's for many years & encountered river crossings so be prepared or you will be pushing.
Here is one of me on my RMX250.