Correction: Towards a seamlessly diagnosable expression for the energy flux associated with both equatorial and mid-latitude waves (Progress in Earth and Planetary Science, 2026)