Please update the test based on the comments above.