Fix tests :-)