FIXES bug in plot_component

Incorrect variable name was used in the function