You can take the contributions any time you want ... you will get a 1099-R reporting the distribution and you will enter it in the program and simply answer the follow up questions to indicate it was a return of contributions ... it is just that simple.
