Improve getIcon method

btut · May 12, 2020 · 478ee05 · 478ee05
1 parent e24c141
commit 478ee05
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/main/java/org/jabref/gui/util/BackgroundTask.java b/src/main/java/org/jabref/gui/util/BackgroundTask.java
@@ -257,11 +257,8 @@ public BackgroundTask<V> withInitialMessage(String message) {
         return this;
     }
 
-    public static Node getIcon(Object task) {
-        if (task instanceof Task) {
-            return BackgroundTask.iconMap.getOrDefault(((Task<?>) task).getTitle(), null);
-        }
-        return null;
+    public static <T> Node getIcon(Task<T> task) {
+        return BackgroundTask.iconMap.getOrDefault(task).getTitle(), null);
     }
 
     static class BackgroundProgress {