Update tests for save policy