fix tests :p