update test to reflect the change