fix tests (oops)