Remove the TestUnit code dependency