Fixed a bug where an incorrect check was done when plotting the input. Does not have any inpact on the results