BUG: mse monitoring plot not working with multiple branches

adjust code to work properly