diff --git a/agent/site.py b/agent/site.py
index b6d21f19..c9e88e2e 100644
--- a/agent/site.py
+++ b/agent/site.py
@@ -198,6 +198,7 @@ def restore_job(
         database,
         public,
         private,
+        sanitized_config_content,
         skip_failing_patches,
     ):
         files = self.bench.download_files(self.name, database, public, private)
@@ -212,6 +213,11 @@ def restore_job(
                     files["public"],
                     files["private"],
                 )
+
+                if not sanitized_config_content:
+                    self.update_config()
+                else:
+                    self.update_config(sanitized_config_content)
             else:
                 self.restore_files(
                     public_file=files["public"],
diff --git a/agent/web.py b/agent/web.py
index 7f6170b2..6caaf19b 100644
--- a/agent/web.py
+++ b/agent/web.py
@@ -611,6 +611,7 @@ def restore_site(bench, site):
             data["database"],
             data.get("public"),
             data.get("private"),
+            data.get("sanitized_config_content"),
             data.get("skip_failing_patches", False),
         )
     )