Fix unittests.