"Updated the test as per the comments"