Make corresponding change to the tests