diff --git a/classifiers.py b/classifiers.py
index d3a6fe6..f190670 100644
--- a/classifiers.py
+++ b/classifiers.py
@@ -57,7 +57,7 @@ class YOLOClassify:
     def classify(self, yolo_classifier_model, image_to_classify):
         image_to_classify = image_to_classify.permute(0, 3, 1, 2)
 
-        if yolo_classifier_model.fp16:
+        if yolo_classifier_model.model.fp16:
             image_to_classify = image_to_classify.half()
         else:
             image_to_classify = image_to_classify.float()