Fixed arguments bug for plotting