Add unittest module back in.