oops! forgot to update tests!