1 files changed, 3 insertions, 2 deletions
diff --git a/java/util/TreeSet.java b/java/util/TreeSet.java
index df8235227..ba852131a 100644
--- a/java/util/TreeSet.java
+++ b/java/util/TreeSet.java
@@ -267,11 +267,12 @@ public class TreeSet extends AbstractSet
   private void writeObject(ObjectOutputStream out) throws IOException
   {
     Iterator itr = map.keySet().iterator();
+    int size = map.size();
 
     out.writeObject(map.comparator());
-    out.writeInt(map.size());
+    out.writeInt(size);
 
-    while (itr.hasNext())
+    for (int i = 0; i < size; i++)
       out.writeObject(itr.next());
   }