unittests fix