update tests as well