Fix units in plotting