fix in plotting