diff --git a/core/agents/spec_writer.py b/core/agents/spec_writer.py
index 9699fa255..ef67b7243 100644
--- a/core/agents/spec_writer.py
+++ b/core/agents/spec_writer.py
@@ -109,7 +109,8 @@ async def update_spec(self, iteration_mode) -> AgentResponse:
             default="yes",
             buttons_only=True,
         )
-        await self.ui.close_diff()
+        if hasattr(self.ui, "close_diff"):
+            await self.ui.close_diff()
 
         if user_response.button == "yes":
             self.next_state.specification = self.current_state.specification.clone()