fix(tests): fixing tests