updating tests so they reflect the new changes