Fixes unittest.