diff --git a/app/src/main/java/org/thoughtcrime/securesms/database/DatabaseObserver.java b/app/src/main/java/org/thoughtcrime/securesms/database/DatabaseObserver.java index ad989fb78..4de69505a 100644 --- a/app/src/main/java/org/thoughtcrime/securesms/database/DatabaseObserver.java +++ b/app/src/main/java/org/thoughtcrime/securesms/database/DatabaseObserver.java @@ -117,6 +117,7 @@ public final class DatabaseObserver { public void unregisterObserver(@NonNull MessageObserver listener) { executor.execute(() -> { messageUpdateObservers.remove(listener); + unregisterMapped(messageInsertObservers, listener); }); } @@ -227,8 +228,8 @@ public final class DatabaseObserver { map.put(key, listeners); } - private void unregisterMapped(@NonNull Map> map, @NonNull Observer listener) { - for (Map.Entry> entry : map.entrySet()) { + private void unregisterMapped(@NonNull Map> map, @NonNull V listener) { + for (Map.Entry> entry : map.entrySet()) { entry.getValue().remove(listener); } }