Transformation policy tests fix