Fixed unittests.