oops fix tests