diff --git a/models/yolo.py b/models/yolo.py
index fb7f1eca..dcb073ee 100644
--- a/models/yolo.py
+++ b/models/yolo.py
@@ -437,7 +437,7 @@ class Model(nn.Module):
                 self.traced=False
 
             if self.traced:
-                if isinstance(m, Detect) or isinstance(m, IDetect) or isinstance(m, IAuxDetect):
+                if isinstance(m, Detect) or isinstance(m, IDetect) or isinstance(m, IAuxDetect) or isinstance(m, IKeypoint):
                     break
 
             if profile: