updated the test to reflect the change