When you report your 1099-R for the 401 withdrawal you will have the opportunity to tell the program that it was reinvested in a retirement account within the sixty day period.
