Morning Veigar
Welcome to the Forum
The box in your car should be 5 speed ZF A5S440Z it may be cheaper to service/repair the box than look for a replacement. When was the box last serviced? ZF advise service every 60 to 100k (kilometres)
If you were thinking of an upgrade there is a lot of work including ECU matching
Good Luck
Dave