Fixed little bug in plotting