issue in YOLOv8Detector

[roykev]

hi,
I'm running the detector on a custom model (i.e., i provide the the weights and classes)
got an error in line 107
warnings.warn(
f"class {_class} not found in model classes list.")

I can resolve it when I replace line 99
class_names = get_names()
with
class_names = list(self.model.names.values())

Am I right?
thanks