diff --git a/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java b/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java
index 0e1f4f6744a4f9e456ef317e6ff356da3d40b8ad..45519be6996ae0926a2b4ee0d0f0d783434b7775 100644
--- a/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java
+++ b/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java
@@ -48,11 +48,11 @@ public class DeepLearningVersionSelector extends EzPlug implements EzStoppable,
      * @return map to let naive users select the wanted DL version
      */
     private static Map<String, DeepLearningVersion> getDisplayMapWithoutPythonRepeated() {
-    	List<DeepLearningVersion> allVersions = AvailableEngines.getForCurrentOS().getVersions();
+    	List<DeepLearningVersion> allVersions = AvailableEngines.getForCurrentOS();
     	List<DeepLearningVersion> vList = AvailableEngines.removeRepeatedPythonVersions(allVersions);
 	    Map<String, DeepLearningVersion> map = vList.stream()
 		        .collect(Collectors.toMap(
-		                v -> v.getEngine() + "-" + v.getPythonVersion()
+		                v -> v.getFramework() + "-" + v.getPythonVersion()
 		                + (v.getCPU() ? "-" + DeepLearningVersion.cpuKey : "")
 		                + (v.getGPU() ? "-" + DeepLearningVersion.gpuKey : ""),
 		                Function.identity()));