diff --git a/src/careamics_napari/training_plugin.py b/src/careamics_napari/training_plugin.py
index ba64ddc..e85bf79 100644
--- a/src/careamics_napari/training_plugin.py
+++ b/src/careamics_napari/training_plugin.py
@@ -4,6 +4,7 @@
 from queue import Queue
 from typing import TYPE_CHECKING, Optional
 
+from careamics import CAREamist
 from careamics.config.support import SupportedAlgorithm
 from qtpy.QtCore import Qt
 from qtpy.QtWidgets import QHBoxLayout, QStackedWidget, QVBoxLayout, QWidget
@@ -42,7 +43,6 @@
 
 if TYPE_CHECKING:
     import napari
-    from careamics import CAREamist
 
 # at run time
 try: