P0137 & p0138 o2 sensor codes and traction Control light

2014 tacoma 2.7 prerunner 150k miles
Last year my front driveshaft yoke broke and messed up a bunch of stuff. o2 sensor codes, traction control light and "limp mode". Fixed the limp mode, a fuse blew which was replaced. As for the o2 sensor codes, I replaced the o2 sensor that sheared off due to broken driveshaft, cleared the codes and it pops back up immediately within a few minutes if that. Along with P0137 and p0138 I still have my traction Control light on. Could it be related and any ideas to get rid of the codes? Also for the last week I've been having a problem with the truck shutting off out of no where. All grounds seem to be tight and clean, no loose connectors, all fuses are still in good shape with nothing blown.

 

Did you closely inspect the wiring harnesses in the area for damage, was the correct OE sensor installed? (Upstream and downstream are different)
Traction light will generally be on with an engine code so the engine code needs to be fixed first.