Updated unittest.