I got it working by using an Android App link from the success url (new version of deeplinks).
So it returns to the app on success.
The only problem is it leaves the checkout window open in the browser with all the credit card details.
Not ideal.
I guess I could use the success url to go to a success webpage and then open then app automatically from there.
Not an ideal solution. I guess Elements is the best way to go.