Dying after killing monolith boss resets your monolith

Intended behaviour:
Dying in a Echo Quest should make you lose you a bit of Stability and you should be able to attempt the quest again
Current behaviour:
If you die to a DoT or ground effect after the final boss is dead the entire monolith progression is reset but you don’t get to choose new monolith unlock. Game treats it as if you just completed the timeline succesfully altough you died.