I re-loaded an old save to have a look at the rewritten scene. I think this is a great improvement. The first one felt like a cheap death that came out of nowhere, but the new one feels more like a natural part of the plot puzzle and motivates you to try again instead.