They likely would have needed to redesign all the controls and figure out the initiation sequences out again from scratch. The weapons in question were fitted with encrypted detonation controls developed with US assistance.

They have a lot of the Soviet weapons design bureaus. Not sure how many of the original designers are still around. The tricky bit will be refining enough uranium or plutonium in a war zone.