The node you are looking for is at plot_shock_decomposition.