Engine Shut Down when Switching 4wd

I have a 21’ Rmax4 XTR with 4,000 miles on it and love every minute riding it.

Last weekend I was riding in WV and the machine would shut down whenever I would switch the 4wd. Happen from 2 wheel to 4 wheel and 4 wheel to 2 wheel. It cut off on me going down a hill coasting too. I noticed the yellow “check engine” light would come on after the engine turned off. I would have to turn the key off and crank it up. It would crank like normal.

I’m curious if this has happened to anyone else and if there is a fix I can do myself or take it to the dealership?

Other than terrible rear axles I haven’t had any issues for 4K miles.

Any suggestions or feedback would be much appreciated. Thank you in advance!