Updated the test method