fix tests bug