Improved energy compensation for time-of-flight mass spectrometry