Fixed a plotting issue