diff --git a/src/main/java/org/bioimageanalysis/icy/deeplearning/versionmanager/downloading/DeepLearningVersionDownloader.java b/src/main/java/org/bioimageanalysis/icy/deeplearning/versionmanager/downloading/DeepLearningVersionDownloader.java
index 22a347593160bee6a6c6142929a38d2aa4af716b..88f343700b50bc14c1bc6da45d168b5926cf977b 100644
--- a/src/main/java/org/bioimageanalysis/icy/deeplearning/versionmanager/downloading/DeepLearningVersionDownloader.java
+++ b/src/main/java/org/bioimageanalysis/icy/deeplearning/versionmanager/downloading/DeepLearningVersionDownloader.java
@@ -23,12 +23,12 @@ import java.util.stream.Collectors;
 import java.util.stream.IntStream;
 
 import org.bioimageanalysis.icy.deeplearning.util.StreamUtils;
-import org.bioimageanalysis.icy.deeplearning.versionmanagement.DeepLearningVersion;
-import org.bioimageanalysis.icy.deeplearning.versionmanagement.InstalledDeepLearningVersions;
 
 import com.google.common.util.concurrent.AtomicDouble;
 
 import icy.common.listener.ProgressListener;
+import io.bioimage.modelrunner.versionmanagement.DeepLearningVersion;
+import io.bioimage.modelrunner.versionmanagement.InstalledDeepLearningVersions;
 
 /**
  * Downloader for Deep Learning engine versions.
diff --git a/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningDownloader.java b/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningDownloader.java
index d28052157f734864c359cff70f20c9f678a076dc..bfcd236daee8bd4d483aed7183d3a22721fc3115 100644
--- a/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningDownloader.java
+++ b/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningDownloader.java
@@ -3,12 +3,12 @@ package plugins.danyfel80.deeplearningdownloader;
 import java.io.IOException;
 import java.util.Objects;
 
-import org.bioimageanalysis.icy.deeplearning.versionmanagement.DeepLearningVersion;
 import org.bioimageanalysis.icy.deeplearning.versionmanager.downloading.DeepLearningVersionDownloader;
 
 import icy.common.listener.ProgressListener;
 import icy.plugin.abstract_.Plugin;
 import icy.plugin.interface_.PluginLibrary;
+import io.bioimage.modelrunner.versionmanagement.DeepLearningVersion;
 
 /**
  * This plugin handles the loading into the JVM of a SINGLE instance of the TensorFlow API for Java.
diff --git a/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java b/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java
index 80f2934f9d2d587a9a4eca7c225b72cb3bef322d..8562fc130f44ab807f4aebf76ba7c0fc277ecbb6 100644
--- a/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java
+++ b/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java
@@ -6,10 +6,6 @@ import java.util.Map;
 import java.util.function.Function;
 import java.util.stream.Collectors;
 
-import org.bioimageanalysis.icy.deeplearning.system.PlatformDetection;
-import org.bioimageanalysis.icy.deeplearning.utils.Constants;
-import org.bioimageanalysis.icy.deeplearning.versionmanagement.AvailableDeepLearningVersions;
-import org.bioimageanalysis.icy.deeplearning.versionmanagement.DeepLearningVersion;
 
 import icy.gui.dialog.MessageDialog;
 import icy.main.Icy;
@@ -17,6 +13,10 @@ import icy.plugin.PluginLauncher;
 import icy.plugin.PluginLoader;
 import icy.preferences.XMLPreferences;
 import icy.util.StringUtil;
+import io.bioimage.modelrunner.system.PlatformDetection;
+import io.bioimage.modelrunner.utils.Constants;
+import io.bioimage.modelrunner.versionmanagement.AvailableDeepLearningVersions;
+import io.bioimage.modelrunner.versionmanagement.DeepLearningVersion;
 import plugins.adufour.blocks.lang.Block;
 import plugins.adufour.blocks.util.VarList;
 import plugins.adufour.ezplug.EzPlug;