fix tests too :-)