update test to match newest code