fix tests ?