fix(tests): unit tests fix