.. fix tests