fix broken tests

Please don't merge PRs without running tests!!