Fixed a plotting error