Fixing unittests.