fix tests, geez