You're not just limited to mk2 gearboxes. You can use mk1 Golf, Scirocco, Jetta and even Austin Maestro

One difference that you do want to look out for is the reversing light connector.......these can be different. You may also need to swap the drive shaft flanges over.
The ratio's can also be a little different depending on what car it came from. I expect the code for your box is: 020 AUG
Have a look at this table so that you can get an idea of the differences:-
