Excellent article re ECU flash tuning in the July issue of Sport Rider, which covers a company called Flash-Tune in Cali. I don't see KTM covered yet but it's the way I would go for a ECU reflash.
https://ftecu.com/
Seems you don't have much choice but to let the shop try and make it right.