In the retirement entry section ... after you enter the 1099-R you need to continue thru all the following screens to indicate your cost basis. If you make a new contribution then you should enter that first and then enter the 1099-R.
