trying to fix tests suit