Ah ok, it was too tight, I got a pen knife and cut off the top bit of the cooler, around 5mm, used to hold the shroud, and now goes in the case with no problem.
It seems there are two versions of the AMD Wraith fan that come with the cpu, mine has a 2mm gap between the fan and heatsink which...