Fixing tests we broke

- oops