diff --git a/src/scportrait/pipeline/segmentation/workflows.py b/src/scportrait/pipeline/segmentation/workflows.py
index 2ad5d0cb..b677597f 100644
--- a/src/scportrait/pipeline/segmentation/workflows.py
+++ b/src/scportrait/pipeline/segmentation/workflows.py
@@ -15,7 +15,6 @@
 from skimage.filters import median
 from skimage.morphology import binary_erosion, dilation, disk, erosion
 from skimage.segmentation import watershed
-import _multiprocessing
 
 from scportrait.pipeline._utils.segmentation import (
     contact_filter,
@@ -1355,7 +1354,7 @@ def _check_gpu_status(self):
             cpu_id = int(cpu_name[cpu_name.find("-") + 1 :]) - 1
 
             if cpu_id >= len(gpu_id_list):
-                cpu_id = cpu_id%current.n_processes
+                cpu_id = cpu_id % current.n_processes
 
             # track gpu_id and update GPU status
             self.gpu_id = gpu_id_list[cpu_id]