Updated test for restore_content