Umbra's new augment is bugged: regaining sentience after Last Gasp revive

[Gallerine]

I was super excited to see an augment added to address the "Umbra + Helminth" situation, however after a short while of gameplay I noticed after a self-revive using the Last Gasp Vazarin skill, that Umbra was fighting alongside me again and effectively disabling the augment.
 

Full context in case clipped video upload fails somehow: After not paying attention and completing some challenges for Onos evolution in the Labs, I died and started the Last Gasp revive timer, upon self-revive I hopped back out of my frame to clean up some enemies since I needed to level an Operator Amp, however during this time I noticed Umbra was following me and shooting behind him as sentience had fully re-enabled.

Link to short clip, beginning as the Last Gasp revive animation plays. (sorry, I don't know how to embed videos directly into forums post)

As for other details:
I can guarantee that I had the augment equipped during this mission as was on previous missions beforehand. Desktop PC, solo, good connection, the only other augment equipped was Chromatic Blade.

Hopefully this helps in tracking and fixing this, a bug on the augment implemented to bandaid another bug is rather ironic...