Fixing retry policy tests