diff --git a/commons/src/main/java/lcsb/mapviewer/common/comparator/EnumComparator.java b/commons/src/main/java/lcsb/mapviewer/common/comparator/EnumComparator.java
index 8157cfb562a98b4fce1ac7f5cde1519f95fc24f3..55c899d983f9c772c795842ef6ff38ec2dd15fe8 100644
--- a/commons/src/main/java/lcsb/mapviewer/common/comparator/EnumComparator.java
+++ b/commons/src/main/java/lcsb/mapviewer/common/comparator/EnumComparator.java
@@ -9,10 +9,10 @@ import java.util.Comparator;
  * @author Piotr Gawron
  * 
  */
-public class EnumComparator implements Comparator<Enum<?>> {
+public class EnumComparator<E extends Enum<E>> implements Comparator<E> {
 
 	@Override
-	public int compare(Enum arg0, Enum arg1) {
+	public int compare(E arg0, E arg1) {
 		if (arg0 == null) {
 			if (arg1 == null) {
 				return 0;
diff --git a/commons/src/test/java/lcsb/mapviewer/common/comparator/EnumComparatorTest.java b/commons/src/test/java/lcsb/mapviewer/common/comparator/EnumComparatorTest.java
index 58c410181b3c25facae0b2d24553fdc3c08b1e7f..dd8638cd527a00617f59ea0771c1d7a6afa2486c 100644
--- a/commons/src/test/java/lcsb/mapviewer/common/comparator/EnumComparatorTest.java
+++ b/commons/src/test/java/lcsb/mapviewer/common/comparator/EnumComparatorTest.java
@@ -11,13 +11,13 @@ public class EnumComparatorTest {
 
   @Test
   public void testEquals() {
-    EnumComparator comp = new EnumComparator();
+    EnumComparator<MimeType> comp = new EnumComparator<>();
     assertEquals(0, comp.compare(MimeType.CSS, MimeType.CSS));
   }
 
   @Test
   public void testDifferent() {
-    EnumComparator comp = new EnumComparator();
+    EnumComparator<MimeType> comp = new EnumComparator<>();
     assertTrue(0 != comp.compare(MimeType.CSS, MimeType.JPG));
   }
 
diff --git a/model/src/main/java/lcsb/mapviewer/model/map/kinetics/SbmlUnitTypeFactorComparator.java b/model/src/main/java/lcsb/mapviewer/model/map/kinetics/SbmlUnitTypeFactorComparator.java
index c585a94b81b67e8ae7f04697f9d9442ea5fa5d12..c54e7745fe33855e24b1df4a1822e2924a9a9916 100644
--- a/model/src/main/java/lcsb/mapviewer/model/map/kinetics/SbmlUnitTypeFactorComparator.java
+++ b/model/src/main/java/lcsb/mapviewer/model/map/kinetics/SbmlUnitTypeFactorComparator.java
@@ -15,7 +15,7 @@ public class SbmlUnitTypeFactorComparator extends Comparator<SbmlUnitTypeFactor>
   protected int internalCompare(SbmlUnitTypeFactor arg0, SbmlUnitTypeFactor arg1) {
     IntegerComparator integerComparator = new IntegerComparator();
     DoubleComparator doubleComparator = new DoubleComparator();
-    EnumComparator enumComparator = new EnumComparator();
+    EnumComparator<SbmlUnitType> enumComparator = new EnumComparator<>();
     if (integerComparator.compare(arg0.getExponent(), arg1.getExponent()) != 0) {
       return integerComparator.compare(arg0.getExponent(), arg1.getExponent());
     }
diff --git a/model/src/main/java/lcsb/mapviewer/model/map/species/SpeciesComparator.java b/model/src/main/java/lcsb/mapviewer/model/map/species/SpeciesComparator.java
index 539b1d8b9fb85ecc1f30abf4ca50879e5a03a1e3..fccb26a474ad50d5396e390e57de40a9b107490d 100644
--- a/model/src/main/java/lcsb/mapviewer/model/map/species/SpeciesComparator.java
+++ b/model/src/main/java/lcsb/mapviewer/model/map/species/SpeciesComparator.java
@@ -1,6 +1,7 @@
 package lcsb.mapviewer.model.map.species;
 
-import org.apache.logging.log4j.*;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
 
 import lcsb.mapviewer.common.Comparator;
 import lcsb.mapviewer.common.Configuration;
@@ -9,6 +10,7 @@ import lcsb.mapviewer.common.comparator.DoubleComparator;
 import lcsb.mapviewer.common.comparator.EnumComparator;
 import lcsb.mapviewer.common.comparator.IntegerComparator;
 import lcsb.mapviewer.common.comparator.StringComparator;
+import lcsb.mapviewer.model.map.kinetics.SbmlUnitType;
 
 /**
  * Comparator class used for comparing {@link Species} objects.
@@ -147,7 +149,7 @@ public class SpeciesComparator extends Comparator<Species> {
       return booleanComparator.compare(arg0.isBoundaryCondition(), arg1.isBoundaryCondition());
     }
 
-    EnumComparator enumComparator = new EnumComparator();
+    EnumComparator<SbmlUnitType> enumComparator = new EnumComparator<>();
     if (enumComparator.compare(arg0.getSubstanceUnits(), arg1.getSubstanceUnits()) != 0) {
       logger.debug(
           "SubstanceUnits different: \"" + arg0.getSubstanceUnits() + "\", \"" + arg1.getSubstanceUnits() + "\"");