[#65157742] Adjusted code to conform to the test case