Fix stupid bug in unittest.