UPDATE:
It seems to be a visual bug only, as the Step 2 also disappears when you click Back.
@shane90 Once you have added the records in your registrar, click Verify (I clicked several times but I am sure once is enough ), close the wizard and wait.
Eventually, the Domain tab of the Share popup still shows the custom domain so it is still working in the backend.
Hope that helps