attempt to fix tests