Have you re-read the OBD codes since the repair?

There may be some other problem occurring.

The heated sensor is powered through a fuse, so if the fuse has failed then it might trigger the same trouble code as a bad sensor.

It makes some sense that the 'upstream' oxygen sensor would be on the exhaust system very close to the engine and may well be the part visible when looking at the engine from the open hood.