You can only do that on the 2020 tax return to spread the withdrawal from the 401(k) over three years due to Covid.
