Retirement tax questions

Can I transfer the money now back to my savings account while waiting for the IRA custodian to make the “return of contribution” form?  Or should I wait for them to make the form first?