fix(tests): Adjusted tests to pass properly