Bug fix in unittest.