fix for unittests