diff --git a/sets/README.md b/sets/README.md
index 7e12243..12aa549 100644
--- a/sets/README.md
+++ b/sets/README.md
@@ -32,8 +32,8 @@ class RandomizedSet:
 
     def remove(self, val: int) -> bool:
         if val in self.dict:
-            index = self.dict[val]
-            self.set[index] = self.set[-1]
+            last_element, idx = self.set[-1], self.dict[val]
+            self.set[idx], self.dict[last_element] = last_element, idx
             self.set.pop()
             del self.dict[val]
             return True