2 files changed, 7 insertions, 2 deletions

diff --git a/2025/08/a.py b/2025/08/a.py
index b829dca..6201d09 100644
--- a/2025/08/a.py
+++ b/2025/08/a.py
@@ -7,7 +7,10 @@ class SetJoin:
                 self._parent = [i for i in range(n)]
 
         def rep(self, i):
-                return i if self._parent[i] == i else self.rep(self._parent[i])
+                if self._parent[i] != i:
+                        self._parent[i] = self.rep(self._parent[i])
+                return self._parent[i]
+
 
         def join(self, i, j):
                 self._parent[self.rep(i)] = self.rep(j)
diff --git a/2025/08/b.py b/2025/08/b.py
index 0fac672..eb171f9 100644
--- a/2025/08/b.py
+++ b/2025/08/b.py
@@ -5,7 +5,9 @@ class SetJoin:
                 self._parent = [i for i in range(n)]
 
         def rep(self, i):
-                return i if self._parent[i] == i else self.rep(self._parent[i])
+                if self._parent[i] != i:
+                        self._parent[i] = self.rep(self._parent[i])
+                return self._parent[i]
 
         def join(self, i, j):
                 self._parent[self.rep(i)] = self.rep(j)