Oops, fixed tests again