diff --git a/models.py b/models.py
index 9ce72ece..ae861cec 100755
--- a/models.py
+++ b/models.py
@@ -206,7 +206,7 @@ class YOLOLayer(nn.Module):
             if self.nc == 1:
                 io[..., 5] = 1  # single-class model https://github.com/ultralytics/yolov3/issues/235
 
-            # reshape from [1, 3, 13, 13, 85] to [1, 507, 84], remove obj_conf
+            # reshape from [1, 3, 13, 13, 85] to [1, 507, 85]
             return io.view(bs, -1, self.no), p