diff --git a/pyproject.toml b/pyproject.toml
index ba715b4..8bcd8e0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "tools"
-version = "0.1.3"
+version = "0.1.4"
 description = "Converter for YOLO models into .ONNX format."
 readme = "README.md"
 requires-python = ">=3.8"
diff --git a/tools/modules/heads.py b/tools/modules/heads.py
index 81cb81c..664af3c 100644
--- a/tools/modules/heads.py
+++ b/tools/modules/heads.py
@@ -482,7 +482,6 @@ def __init__(self, old_segment, use_rvc2):
         self.nm = old_segment.nm  # number of masks
         self.npr = old_segment.npr  # number of protos
         self.proto = old_segment.proto  # protos
-        self.detect = old_segment.detect
         self.cv4 = old_segment.cv4
 
     def forward(self, x):