HI Neil,
Really sorry to hear about your gearbox - what a bummer!
Good thing you've still got the A4 so you're atleast still mobile.
When my 3.7 engine went, it was a real pain finding a replacement head for it - i hope you have better luck if an exchange it the route you're going down mate.
Fingers crossed you get it sorted soon!