fix tests breaks