tried to fix tests