Fix retention policy tests