diff --git a/src/utils.rs b/src/utils.rs
index 8f4437f..3518359 100644
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -2,9 +2,7 @@ use tch::{IValue, Tensor};
 
 use crate::{image::ImageCHW, BBox, SegBBox, SegmentationResult};
 
-pub struct SegmentationTools {
-    device: tch::Device,
-}
+pub struct SegmentationTools {}
 pub struct DetectionTools {}
 
 impl SegmentationTools {