Yes - double check the worksheet but sounds right to me.
Stimulus 1 = $2,400
Stimulus 2 = $1,200
Less received $2,206 = $1,394.
The key is the wording from the IRS "The first two rounds of Economic Impact Payments were advance payments of the 2020 Recovery Rebate Credit." your 2020 return shows eligibility based on your 2020 income.