fix label of energy plot