diff --git a/cfg/yolov7.yaml b/cfg/yolov7.yaml
index 6def0f2..dae5e56 100644
--- a/cfg/yolov7.yaml
+++ b/cfg/yolov7.yaml
@@ -132,5 +132,5 @@ head:
    [88, 1, Conv, [512, 3, 1]],
    [101, 1, Conv, [1024, 3, 1]],
 
-   [[102, 103, 104], 1, V6Detect, [nc]],  # Detect(P3, P4, P5)
+   [[102, 103, 104], 1, IV6Detect, [nc]],  # Detect(P3, P4, P5)
   ]