Stage music for PC AE

I know there’s a pack of bgm files for character themes, but what about stage music? I’ve messed around with the ones already in the install folder, but it seems the regular music and the Ultra/ heavy damage music is on one file, and if you use training mode, it will switch to the Ultra variation and stay there. I don’t know of any convenient way around that.