Fixed unittest.