X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FReplyGroupFilter.java;h=709ffbc26f0834386abbeb5bcd638ba2e3544b1b;hp=bcf73d8ab97371b881479b36bf533e0b2481cb18;hb=419098bcd6215125408b29e60bd888e60979d37b;hpb=6d8f11098552fcb40a87d3e35690b3ba760fb947

diff --git a/src/main/java/net/pterodactylus/sone/template/ReplyGroupFilter.java b/src/main/java/net/pterodactylus/sone/template/ReplyGroupFilter.java
index bcf73d8..709ffbc 100644
--- a/src/main/java/net/pterodactylus/sone/template/ReplyGroupFilter.java
+++ b/src/main/java/net/pterodactylus/sone/template/ReplyGroupFilter.java
@@ -1,5 +1,5 @@
 /*
- * Sone - ReplyGroupFilter.java - Copyright © 2010–2013 David Roden
+ * Sone - ReplyGroupFilter.java - Copyright © 2010–2015 David Roden
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -21,6 +21,7 @@ import java.util.HashMap;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Map.Entry;
 import java.util.Set;
 
 import net.pterodactylus.sone.data.Post;
@@ -69,14 +70,14 @@ public class ReplyGroupFilter implements Filter {
 			replies.add(reply);
 		}
 		Map<Post, Map<String, Set<?>>> result = new HashMap<Post, Map<String, Set<?>>>();
-		for (Post post : postSones.keySet()) {
-			if (result.containsKey(post)) {
+		for (Entry<Post, Set<Sone>> postEntry : postSones.entrySet()) {
+			if (result.containsKey(postEntry.getKey())) {
 				continue;
 			}
 			Map<String, Set<?>> postResult = new HashMap<String, Set<?>>();
-			postResult.put("sones", postSones.get(post));
-			postResult.put("replies", postReplies.get(post));
-			result.put(post, postResult);
+			postResult.put("sones", postEntry.getValue());
+			postResult.put("replies", postReplies.get(postEntry.getKey()));
+			result.put(postEntry.getKey(), postResult);
 		}
 		return result;
 	}