Update the test to match the fix from yesterday