Someone has already tried the AE gear diff and it doesn't fit in a X2c gearbox. The original B4 diff is smaller than the B4.1 diff which is the same size as the gear diff used in the SC10/ b4.1 I believe.
Just have to hold tight for the team c one, personally I haven't had many problems with the ball one though.