fix unit tests :-)