fix unittests