Add unittest back.