Bugfix for plotting.