feat(tests): updated tests