diff --git a/src/napari_tardis_em/viewers/viewer_predict.py b/src/napari_tardis_em/viewers/viewer_predict.py
index 65b04e5..3e20a1f 100644
--- a/src/napari_tardis_em/viewers/viewer_predict.py
+++ b/src/napari_tardis_em/viewers/viewer_predict.py
@@ -8,7 +8,6 @@
 #  MIT License 2024                                                   #
 #######################################################################
 from os import getcwd
-from os.path import join
 
 import numpy as np
 import torch
diff --git a/src/napari_tardis_em/viewers/viewer_utils.py b/src/napari_tardis_em/viewers/viewer_utils.py
index f07468d..a199912 100644
--- a/src/napari_tardis_em/viewers/viewer_utils.py
+++ b/src/napari_tardis_em/viewers/viewer_utils.py
@@ -1,18 +1,15 @@
 from os.path import join, dirname
 
 import numpy as np
+
 from napari.utils.notifications import show_error
-from tardis_em.cnn.data_processing.trim import trim_with_stride
+from napari_tardis_em.viewers.utils import create_image_layer, create_point_layer
 
+from tardis_em.cnn.data_processing.trim import trim_with_stride
 from tardis_em.cnn.datasets.dataloader import PredictionDataset
-
 from tardis_em.utils.predictor import GeneralPredictor
-
-from napari_tardis_em.viewers.utils import create_image_layer, create_point_layer
 from tardis_em.utils.aws import get_all_version_aws
-
 from tardis_em.analysis.filament_utils import sort_by_length
-
 from tardis_em.utils.normalization import adaptive_threshold