fix tests fail