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.