Adjusted test to match the expected outcome