Added the wait step and made it so checking for the PIN screen wouldn't fail.