Tests have been fixed to reflect changes made by pundit