trying to fix tests