Update the test for clarity