After you file

Did you look at the form 1040 lines 5a & 5b ... the entire distribution will be listed on line 5a and the taxable portion on line 5b.  The summary in the program only shows 5a which can be confusing  ...  so if the return is incorrect  amending it would be the way to go.