Use “isLoaded” method to check for not-yet loaded bookmarks.

diff --git a/src/main/java/net/pterodactylus/sone/web/BookmarksPage.java b/src/main/java/net/pterodactylus/sone/web/BookmarksPage.java
index 373ff18..0f53572 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/web/BookmarksPage.java
+++ b/src/main/java/net/pterodactylus/sone/web/BookmarksPage.java
@@ -68,7 +68,7 @@ public class BookmarksPage extends SoneTemplatePage {
 
                        @Override
                        public boolean apply(Post post) {
-                               return post.getSone() != null;
+                               return post.isLoaded();
                        }
                });
                List<Post> sortedPosts = new ArrayList<Post>(loadedPosts);

diff --git a/src/main/java/net/pterodactylus/sone/web/UnbookmarkPage.java b/src/main/java/net/pterodactylus/sone/web/UnbookmarkPage.java
index 7dfe301..72ff2fc 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/web/UnbookmarkPage.java
+++ b/src/main/java/net/pterodactylus/sone/web/UnbookmarkPage.java
@@ -67,7 +67,7 @@ public class UnbookmarkPage extends SoneTemplatePage {
                if (id.equals("allNotLoaded")) {
                        Set<Post> posts = webInterface.getCore().getBookmarkedPosts();
                        for (Post post : posts) {
-                               if (post.getSone() == null) {
+                               if (post.isLoaded()) {
                                        webInterface.getCore().unbookmarkPost(post);
                                }
                        }