Updated the test to reflect the new changes