My Urban Gray P01 Omega switched from decocker to safeties really slick/easy.
When I got my black P01 Omega I had issues with the safeties not wanting to line up/go together. I fiddled with them/it for a bit but got no where.
Then it hit me I had a working Urban Gray P01. I grabbed it, unloaded it, popped the two safeties out and installed them in the black P01 Omega. No sweat, they went right in. I figured there was something wrong with those safeties. The I got another "hit" and grabbed the two from the black P01 lying on the table and they went right into the Urban Gray P01.
Something a little different from one set of safeties or one frame to the other. To this day they "wear" each other's safeties.
Never had the issue on any of the P07's or P09's.
Based on the one time I had an issue I couldn't correct on my own (tritium front sight on the urban gray P01 died) CZ USA took care of it very quickly after a phone call to talk to them about it) I would expect you'll be taken care of by them.