diff --git a/pdf_agents/sklearn.py b/pdf_agents/sklearn.py
index 215968a..b926a69 100644
--- a/pdf_agents/sklearn.py
+++ b/pdf_agents/sklearn.py
@@ -40,6 +40,7 @@ def n_clusters(self):
     @n_clusters.setter
     def n_clusters(self, value):
         self.model.set_params(n_clusters=int(value))
+        self.close_and_restart()
 
     def server_registrations(self) -> None:
         self._register_method("clear_caches")