Did you get it fixed, what was it? I have the exact same problem!
At the last race it went into limp mode and just after I got off the track it turned off and I couldn't get it started anymore. The Dealer ended up replacing the battery, he said the old one didn't have a consistent charge.
So I went to a race this weekend with a new battery, but I could only ride as long as it was charged, which was enough for practise. It ended up dieing on me right at the gate on the first moto, can tell ya it's pretty frustrating to have to watch everybody else drive off and you have to push your bike back to the trailer just because your dealer didn't do his job.
It'd be great if you could post, what was necessary to fix the problem.