fix(tests): fixed failing tests