Wait until you get the Form 1099-R. Enter it in Wages and Income, Retirement, then answer the questions. You took a $55,000 distribution and rolled over (back) $54,000, so will owe taxes on the remaining $1,000, plus possibly an early withdrawal penalty.