adjusted test to reflect new code