Retirement tax questions

I just took the distribution from CompanyB for the shortfall of 1,000 on 3/17/24.  Because this is taken in 2024, for the shortfall in 2023,  it is correct that I should report this 1,000 on Form 5329 for 2023 (I will ask for waiver on the 1,000) , not 2024.