adjust test to correspond changes