Fix tests :_(