Doing some donuts in some 1ft deep puddles. My engine cut off and is throwing out 2 codes. P0335 and P1341. Having a rough time starting. When it does start, its rough and dies unless I hold rpms at 2500 then as soon as I let off the throttle it dies. First thing I did was check the air box and it was dry. Letting it sit to see if some connections dry out. I had my wheels turned drivers side and no splash guards, so it was just throwing water all around engine but mainly drivers side. I unplugged my crankshaft sensor plug and camshaft sensor plug, both seemed dry. Unplugged battery for a bit. Hooked it all back up and it had reset itself while starting and running perfect. I drove it a little over a block and it died and codes are back. Thoughts on what I did and can do to solve it.