Not sure about the rough idle caused by new O2 sensor, but VW issued a software update/ECM software update for the P0134 concern on VR6's like the one you have, I would call a local dealer and ask them to run your VIN and check for an N7/26E1 campaign/ reflash.
Maybe try cleaning your throttle body to clear up your cold start rough idle.






