Fixed the test as per the comment