update tests to pass again