[Test] Fix previous test change as the test is dependent on line numbers