update tests so they pass