Sounds like the fuel pump issue. I hope it's not but you should try your dealer and see if they will help you get a replacement from KTM/Husaberg. Many posts on this subject and there are others who know much more than I do about this. I have the same bike and did a lot of precautionary measures (lined tank with reflective tape, removed ball valve in gas cap, added a fan kit, wrapped the header pipe and did the recall which was an add on in-line fuel filter.
Best of luck...I hope you get it sorted out.
J