oops
fix tests